运行代码时报错:
RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /opt/conda/conda-bld/pytorch-nightly_1553749764730/work/aten/src/THC/THCGeneral.cpp:51
解决办法:
环境没有问题的情况下,可能是使用GPU的编号有问题。
- 新建终端窗口,运行命令sudo nvidia-settings,在弹出的NVIDIA窗口中查看自己的显卡编号,如下图所示
- 打开代码运行的Python源码文件,查找形如os.environ[“CUDA_VISIBLE_DEVICES”]的代码行,将右边的数字改为上述查看的显卡编号,如下图所示:
好了,完成!