cuda - ValueError: Unknown CUDA arch (8.6) or GPU not supported
问题描述
当我在我的 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
有什么帮助吗?
解决方案
您的 GPU 对于 CUDA 10 来说“太新”了。在 CUDA 11.1中添加了对具有 8.6 计算能力的卡的支持(感谢 @RobertCrovella 的更正)。这意味着您需要使用 CUDA 11.1 或更高版本。
推荐阅读
- android - Android Firebase - 本地缓存数据
- tensorflow-lite - 如何让 Kirin 990 的 NPU 在 TensorFlow Lite 上运行?
- wordpress - 使用 ACF 更新/创建自定义帖子类型后在 Wordpress 中调用函数
- php - 在 Laravel 宏中使用函数
- proxy - Envoy - 重写主机名,保留通配符子域
- mongodb - 将 mongodb atlas 连接到 shell 时出错
- reactjs - 将 ReactJS 和 ExpressJS 部署到 Nginx
- python - django中的数字总和
- scala - 可以将 mlflow.spark 保存的模型加载为 Spark/Scala 管道吗?
- reactjs - 使用函数作为 React 道具的 TypeScript 错误?