ptrblck:

It would be interesting to see which NVIDIA driver is used and how you are executing the command, i.e. are you in a docker or bare metal?


I was running PyTorch example on custom container built over nvidia/cuda:11.1.1 docker. ( Docker Hub 2 )
CUDA, CUDNN, NCCL came with docker.