在使用Paddle
、Pytorch
或者是Tensorflow
的时候,可能出现以下错误
ExternalError: CUDA error(100), no CUDA-capable device is detected.
[Hint: 'cudaErrorNoDevice'. This indicates that no CUDA-capable devices were detected by the installed CUDA driver. ] (at ..\paddle\fluid\platform\gpu_info.cc:108)
[operator < fill_constant > error]
或者是
Compiled with WITH_GPU, but no GPU found in runtime.
说白了就是GPU不可用,研究了很久很久,尝试了各种方法,最后找到了解决方案:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
只需要在代码中添加以上代码即可,需要说明的是,一定要加在文件的最顶部,再引入其他的深度学习库!
以下是正确的使用方法:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
import paddle
以下是错误的使用方法:
import paddle
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'