Using CUDA 11.x but getting error: Unknown CUDA arch (8.6) or GPU not supported
Using CUDA 11.x but getting error: Unknown CUDA arch (8.6) or GPU not supported
我正在设置 conda 环境以使用 pytorch 1.4.0（在 Ubuntu 20.04.2 上），但收到错误消息：
ValueError: Unknown CUDA arch (8.6) or GPU not supported
我知道以前有人问过这个问题，但没有一个答案适合我的情况。 This 回答提示CUDA版本太旧。但是，我将我的 CUDA 版本更新到最新版本，并收到相同的错误消息。
nvcc -V 说我安装了 CUDA 11，当我 运行 nvidia-smi 我得到这个信息：
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.84 Driver Version: 460.84 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
根据 NVIDIA docs，应该兼容：
再补充一个问题：CUDA arch(8.6)中的“8.6”代表什么？
您似乎可以从 here:
获取适用于 CUDA 10.0 的 PyTorch v1.4
pip install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
特定版本的 PyTorch 仅适用于特定版本的 CUDA。
如果您使用的是 CUDA-11.1，则需要最新版本的 PyTorch。您需要升级您的 PyTorch，或者降级您的 CUDA。
我正在设置 conda 环境以使用 pytorch 1.4.0（在 Ubuntu 20.04.2 上），但收到错误消息：
ValueError: Unknown CUDA arch (8.6) or GPU not supported
我知道以前有人问过这个问题，但没有一个答案适合我的情况。 This 回答提示CUDA版本太旧。但是，我将我的 CUDA 版本更新到最新版本，并收到相同的错误消息。
nvcc -V 说我安装了 CUDA 11，当我 运行 nvidia-smi 我得到这个信息：
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.84 Driver Version: 460.84 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
根据 NVIDIA docs，应该兼容：
再补充一个问题：CUDA arch(8.6)中的“8.6”代表什么？
您似乎可以从 here:
获取适用于 CUDA 10.0 的 PyTorch v1.4pip install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
特定版本的 PyTorch 仅适用于特定版本的 CUDA。
如果您使用的是 CUDA-11.1，则需要最新版本的 PyTorch。您需要升级您的 PyTorch，或者降级您的 CUDA。