pytorch - 使用 CUDA 11.x 但出现错误：Unknown CUDA arch (8.6) or GPU not supported

Question

我正在设置一个 conda 环境以使用 pytorch 1.4.0（在 Ubuntu 20.04.2 上），但收到错误消息：

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

我知道以前有人问过这个问题，但没有答案适合我的情况。这个答案表明 CUDA 版本太旧了。但是，我将 CUDA 版本更新为最新版本，并收到相同的错误消息。

nvcc -V 说我安装了 CUDA 11，当我运行 nvidia-smi 时，我得到以下信息：

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

根据NVIDIA docs，它应该兼容：

另一个辅助问题：CUDA拱门（8.6）中的“8.6”代表什么？

标签： pytorchcudanvidia

Answer 1

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

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