Just to be clear, are you referring to the bash script provided alongside the docker image and arguments in the task settings?
Yeah, when doing:
task.set_base_docker( "nvidia/cuda:11.4.1-cudnn8-runtime-ubuntu20.04", docker_setup_bash_script=script, )
So yes, it is supported in the k8s glue
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?