Reputation: 509
I have cuda 11.2 in my PC and want to install PyTorch. PyTorch has only mentions of CUDA10.2 and 11.3 in it's website
Can I install torch==1.10.1+cu113 on my PC? If not, how can I install PyTorch for CUDA11.2
I don't want to change my CUDA version as I have other applications using it.
Upvotes: 5
Views: 28312
Reputation: 773
Below worked fine in my case on CUDA 11.2
conda install -y pytorch==1.12.0 torchvision==0.13.0 -c pytorch
Upvotes: 1
Reputation: 2133
I tried the one for 11.3 and so far it works fine w/ the GPU:
sudo pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
Upvotes: 6
Reputation: 85
sure yes.
uninstall current cuda first
sudo /usr/local/cuda-X.Y/bin/cuda-uninstaller
and use dpkg uninstall libcudnn
ex: sudo dpkg --remove libcudnn7
then download runfile and install it also install corresponding libcudnn version
Upvotes: -1
Reputation: 509
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
This worked but is not the latest version though
Upvotes: 1