PyTorch + CUDA 11.6

Hello everyone,

As a follow-up to this question PyTorch + CUDA 11.4
I have installed these Nvidia drivers version 510.60.02 along with Cuda 11.6.
I’d like to install Pytorch in a conda virtual environment, and I’ve found in the Pytorch website that we couldn’t choose a stable version that relies on the latest versions of Cuda (the older version is 11.3) Start Locally | PyTorch

How can I install/build Pytorch with the latest version of Cuda in my conda virtual env ?

You could build PyTorch from source following these instructions or install the nightly binaries with CUDA11.6 via:

pip install torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
5 Likes

@ptrblck I installed it using the nightly binaries. Thanks a lot for your help !

Thank you so much. Also worked for torchvision.