Is it safe to use PyTorch 1.10 with CUDA 11.1?

If you are installing the pip wheels or conda binaries, then note that they are shipping with their own CUDA runtime (statically linked in the pip wheels, dynamically via cudatoolkit in the conda binaries).
Your local CUDA toolkit will be used, if you are building from source or custom CUDA extensions.

No, since CUDA, cuDNN, NCCL, etc. are linked to the binaries, no kernels should be missing (there was a cuDNN issue with static linking, but that’s independent from the wheels).

1 Like