CODEWITHSUNDEEP
pytorchcudanvidia
JTB
JTB

Reputation: 574

Using CUDA 11.x but getting error: Unknown CUDA arch (8.6) or GPU not supported

I'm setting up a conda environment to use pytorch 1.4.0 (on Ubuntu 20.04.2), but getting the error message:

ValueError: Unknown CUDA arch (8.6) or GPU not supported

I know this has been asked before, but no answer fits my case. This answer suggests that the CUDA version is too old. However, I updated my CUDA version to the most recent, and get the same error message.

nvcc -V says I have CUDA 11 installed, and when I run nvidia-smi I get this info:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.84       Driver Version: 460.84       CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+

which, according to the NVIDIA docs, should work be compatible:

enter image description here

Another auxilliary question: What does the "8.6" in CUDA arch (8.6) represent?

Upvotes: 0

Views: 3535

Answers (2)

MWB
MWB

Reputation: 12567

Specific versions of PyTorch work only with specific versions of CUDA.

If you are using CUDA-11.1, you'll need a fairly recent version of PyTorch. You need to either upgrade your PyTorch, or downgrade your CUDA.

Upvotes: 2

Ivan
Ivan

Reputation: 40638

It seems you can grab PyTorch v1.4 for CUDA 10.0 from here:

pip install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

Upvotes: 1

Related Questions