Just to be clear, are you referring to the bash script provided alongside the docker image and arguments in the task settings?
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?
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