Solved by ptrblck in post #2 
              
              
                The pip wheels and conda binaries ship with their own CUDA runtime (as specified during the installation), so you would only need a corresponding NVIDIA driver to run your code. Your local CUDA toolkit will be used if you are building PyTorch from source or a custom CUDA extension. 
The docker ima…