这篇文章是参考这个博客的:https://blog.csdn.net/ruiying413/article/details/105160331
也是最近在训练数据的时候爆出了这个错误:

RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at ...

然后当时就看到了这篇博客
首先在命令行里敲入(我的是Linux)以下命令:

nvidia-smi

看NVIDIA的型号,我的是:
在这里插入图片描述
刚好和博主的一样,所以需要在py文件里的os.environ[“CUDA_VISIBLE_DEVICES”]的代码行,将右边的数字改为上述查看的显卡编号。

而最主要的就是,怎么找到对应的文件,我是看命令行里写的traceback里找对对应的.py文件
在这里插入图片描述
然后一直找,就找到那个os.environ[“CUDA_VISIBLE_DEVICES”],并改好就行了(并且注意python格式就行了)
在这里插入图片描述

GitHub 加速计划 / li / linux-dash
10.39 K
1.2 K
下载
A beautiful web dashboard for Linux
最近提交(Master分支:2 个月前 )
186a802e added ecosystem file for PM2 4 年前
5def40a3 Add host customization support for the NodeJS version 4 年前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐