Thank you for getting back to me

I have set agent{cuda_version: 11.2}

Thank you

or cu11x

CUDA is the driver itself. The agent doesn't install CUDA but installs a compatible torch assuming that CUDA is properly installed.

If I run nvidia-smi it returns valid output and it says the CUDA version is 11.2

Just to make sure, run the code on the machine itself to verify that python can actually detect the driver

agent.cuda_version="11.2"

Isn't the problem that CUDA 12 is being installed?

Solved that by setting docker_args=["--privileged", "--network=host"]

Hi @<1734020162731905024:profile|RattyBluewhale45> , what version of pytorch are you specifying?