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

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?

Getting the info from the task https://github.com/allegroai/clearml-agent/blob/3c8e0ae5dbf819a8125d0a8b8e70739e1e8b37fe/clearml_agent/glue/k8s.py#L385 , extracting the script https://github.com/allegroai/clearml-agent/blob/3c8e0ae5dbf819a8125d0a8b8e70739e1e8b37fe/clearml_agent/glue/k8s.py#L493 .