Your locally installed CUDA toolkit won’t be used unless you build PyTorch from source or a custom CUDA extension, since the PyTorch binaries (pip wheels and conda binaries) ship with their own CUDA dependencies.
1 Like