当我在 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
有什么帮助吗?
答案 0 :(得分:2)
您的 GPU 对于 CUDA 10 来说“太新了”。对像您这样的 compute capability 8.6 卡的支持是 added in CUDA 11.1(感谢 @RobertCrovella 的更正)。这意味着您需要使用 CUDA 11.1 或更高版本。