Ok, but doesn't work for me though. Can you or AgitatedDove14 help me in linking to relevant code so that I can see what's wrong?
Just to be clear, are you referring to the bash script provided alongside the docker image and arguments in the task settings?
So yes, it is supported in the k8s glue
Yeah, when doing:
task.set_base_docker( "nvidia/cuda:11.4.1-cudnn8-runtime-ubuntu20.04", docker_setup_bash_script=script, )