tensorflow从入门到放弃-0

刚接触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 这个有用,万分感谢

posted @ 2019-03-28 16:58  叫我小李好了  阅读(379)  评论(0编辑  收藏  举报