tensorflow从入门到放弃-0

时间:2023-03-08 20:21:06

刚接触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.****.net/u012348774/article/details/78287060 等于没说,别点进去了,不过我确实看了
https://blog.****.net/qq_23926575/article/details/77722967 这个有用,万分感谢