使用 CUDA 11.x 但出现错误：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使用 CUDA 11.x 但出现错误：Unknown CUDA arch (8.6) or GPU not supported
【发布时间】：2021-11-17 02:30:21
【问题描述】：

我正在设置 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”代表什么？

【问题讨论】：

"pytorch 1.4.0" 为什么？
8.6 指的是 Ampere 系列 GPU 的特定成员，例如任何 RTX 30xx 卡。

标签： pytorch cuda nvidia

【解决方案1】：

特定版本的 PyTorch 仅适用于特定版本的 CUDA。

如果您使用的是 CUDA-11.1，则需要相当新的 PyTorch 版本。您需要升级 PyTorch，或降级 CUDA。

【讨论】：

有没有办法知道哪个 CUDA 版本与我的 pytorch 版本兼容？
@JoshBone 看看我在下面给出的链接。
@JoshBone 是的。 pytorch.org 列出了它们。另请参阅 Ivan 的链接。

【解决方案2】：

看来您可以从here 获取适用于 CUDA 10.0 的 PyTorch v1.4：

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

【讨论】：

我已经安装了 pytorch。这不是我遇到的问题。