I would recommend to uninstall the current pip wheel via pip uninstall torch and then either install the nightly from here 1 (select “Preview (Nightly)”) with CUDA 10.2 or 11.1 or “Stable (1.8.0)” release with CUDA 11.1.



 pumplerod:

Does that mean I would use pip to install CUDA11.1?


The pip wheels and conda binaries ship with their own CUDA runtime (as well as cudnn, NCCL, etc.). You won’t get the complete CUDA toolkit, but could use PyTorch on the GPU with it (except compiling custom CUDA extensions, building PyTorch from source etc.).