刚接触tensorflow一周,感觉还是有点难度的。遇到这么个问题
failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected
不过也没啥大事情,代码还是跑完了,查了一下发现其实是没有指定GPU出的问题,其实也可能出现现存不足的问题。
解决方法如下:
添加这两行代码
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0“
其中0应该是你自己GPU的ID
参考:
https://blog.csdn.net/u012348774/article/details/78287060 等于没说,别点进去了,不过我确实看了
https://blog.csdn.net/qq_23926575/article/details/77722967 这个有用,万分感谢