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

Posted 2023-03-16

【中文标题】ValueError: Unknown CUDA arch (8.6) or GPU not supported

【英文标题】：

【发布时间】：2021-09-30 06:01:21

【问题描述】：

当我在我的 conda 环境中构建 DCNv2 时，我收到了这条消息。

我已经通过 nvidia-smi 检查了 cuda：

Tue Apr  6 20:03:13 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.67       Driver Version: 460.67       CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce RTX 3070    Off  | 00000000:01:00.0  On |                  N/A |
|  0%   45C    P8    17W / 220W |    448MiB /  7979MiB |     23%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A       955      G   /usr/lib/xorg/Xorg                 53MiB |
|    0   N/A  N/A      1555      G   /usr/lib/xorg/Xorg                210MiB |
|    0   N/A  N/A      1690      G   /usr/bin/gnome-shell               61MiB |
|    0   N/A  N/A      3564      G   ...AAAAAAAAA= --shared-files      108MiB |

和 nvcc -V:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

有什么帮助吗？

【问题讨论】：

CUDA 10.1 不支持您的 GPU

【参考方案1】：

您的 GPU 对于 CUDA 10 来说“太新”了。像您的 added in CUDA 11.1 一样支持带有 compute capability 8.6 的卡（感谢 @RobertCrovella 的更正）。这意味着您需要使用 CUDA 11.1 或更高版本。

【讨论】：

我将 CUDA 升级到 11.3.1，但仍然收到有关“CUDA arch (8.6)”的相同错误消息。你知道为什么会这样吗？

@JoshBone 假设这与 pytorch 有关，请记住 pytorch 不使用您系统的 cuda。它安装了它自己的版本以及 pytorch 二进制文件。请参阅 this answer 和 this answer 了解有关如何让 pytorch 与 CUDA 一起工作的更多详细信息。