no CUDA-capable device is detected
这个错误可能由以下几个原因引起:
显卡驱动与 CUDA 驱动不匹配
这种情况推荐用 Ubuntu 自带的驱动更新功能解决。
可以参考我的这篇博客 Ubuntu 安装驱动及 CUDA。
Python 代码问题
在使用 GPU 运算的时候会使用下边这句话来设置使用的 GPU 是哪个:
1 | os.environ["CUDA_VISIBLE_DEVICES"] = "0" |
有的时候只有一块卡,但在代码里设置了 3,就会出现这种错误,改一下就好了。
Gitalk 加载中 ...