Tensorflow没有看到GPU而是使用CPU，为什么会这样？

Question

我正在使用支持GPU的Tensorflow运行Python脚本。但是，该程序似乎无法识别任何GPU并立即开始使用CPU。可能是什么原因造成的？

Answer 1

只是想在讨论中添加张量流可能会因CUDA initialization failure而停止看到GPU，换句话说，tensorflow会检测到GPU，但无法将任何操作发送到其上，所以它回退到CPU。在这种情况下，您应该在日志中看到错误，例如

{{1}}

原因可能是同时使用GPU的不同进程之间的冲突。在这种情况下，我发现让tensorflow工作的最可靠方法是重启机器。在最坏的情况下，重新安装tensorflow和/或NVidia驱动程序。

当GPU突然停止工作时，请参阅one more case。