Pytoroch 出现 no CUDA-capable device is detected :
1. RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:7
一般是由运行的脚本中设置了CUDA_VISIBLE_DEVICES="3"
。可以通过该命令可以看到需要使用编号为3的显卡,关于显卡编号一般是从0开始编号的,第一块编号为0,第二块编号为1。虽然指定使用第二块显卡,不过我的计算机上面只有一块显卡,所以这样会提示上面的报错信息。将显卡指定行改为如下CUDA_VISIBLE_DEVICES="0"
改完可以正常运行了!
2. 关于tensorflow-GPU版本的安装:
使用conda安装:
conda install tensorflow-gpu
自动安装当前最新版本。