这篇文章是参考这个博客的: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格式就行了)