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

可以图形化看到的东西坚决不会用命令行ORZ

原文地址:https://www.cnblogs.com/IGNB/p/10616169.html