Ubuntu 18.04, Python 3.6.9.
I created python environment and install cuda 10.2, then
pip3 install torch==1.11.0+cu102 torchvision==0.12.0+cu102 torchaudio==0.11.0 --extra-index-url whl/cu102
But then I discovered that
NVIDIA GeForce RTX 3060 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
so I uninstalled cuda 10.2 and torch, and installed cuda 11.7.
And now pip3 install --no-cache-dir torch torchvision torchaudio, gives me torch for cuda 10.2
Collecting torch
Downloading files.pythonhosted.org/packages/a4/54/81b1c3c574a1ffde54b0c82ed2a37d81395709cdd5f50e59970aeed5d95e/torch-1.10.2-cp36-cp36m-manylinux1_x86_64.whl (881.9MB)
I tried other options, but without success
pip3 install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url cu117
Collecting torch==2.0.0+cu117
Could not find a version that satisfies the requirement torch==2.0.0+cu117 (from versions: )
No matching distribution found for torch==2.0.0+cu117
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url cu117
Collecting torch==1.13.1+cu117
Could not find a version that satisfies the requirement torch==1.13.1+cu117 (from versions: 1.0.0, 1.0.1, 1.0.1.post2, 1.1.0, 1.2.0, 1.3.0, 1.3.1, 1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2)
No matching distribution found for torch==1.13.1+cu117
The error is expected since your Ampere GPU needs CUDA 11.x.
The posted install commands use an invalid --index-url since you are only specifying cu117 instead of the URL. Refer to the install commands from our website.
Yes, GPU needs CUDA 11.x. The main question is why pip3 install --no-cache-dir torch torchvision torchaudio, gives me torch for cuda 10.2, not torch for cuda 11.x?
I’m new to the site and have a limit on the number of links, so I removed some of them.
The current binaries are not built with CUDA 10.2 anymore, but your environment still picks it up.
I guess it’s because you are using a deprecated Python 3.6 release, which isn’t supported anymore since 23rd Dec 2021, so you might want to update Python.
The command also works for me properly: