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

问题描述

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

when I build DCNv2 in my conda environment,I got this message.

我已经通过 nvidia-smi 检查过 cuda:

I have checked cuda by nvidia-smi:

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

有什么帮助吗?

推荐答案

您的 GPU太新"了适用于 CUDA 10.支持具有 计算能力 8.6 的卡片，就像您的添加CUDA 11.1(感谢@RobertCrovella 的更正).这意味着您需要使用 CUDA 11.1 或更高版本.

Your GPU is "too new" for CUDA 10. Support for cards with compute capability 8.6 like yours was added in CUDA 11.1 (thank you @RobertCrovella for the correction). This means you'll need to use CUDA 11.1 or newer.

这篇关于ValueError: Unknown CUDA arch (8.6) or GPU not supported的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！