Tensorflow沒有看到GPU而是使用了CPU，怎麼樣？

Question

我使用支持GPU的Tensorflow運行Python腳本。但是，該程序似乎無法識別任何GPU並立即開始使用CPU。這可能是什麼原因？

Answer 1

只是要添加到tensorflow 的討論可能會停止看到了GPU由於CUDA initialization failure，換句話說，tensorflow檢測GPU，但不能派任何運到它，所以它回落到CPU。在這種情況下，你應該會看到一個錯誤日誌中，像這樣的

E tensorflow/stream_executor/cuda/cuda_driver.cc:481] failed call to cuInit: CUDA_ERROR_UNKNOWN 

的原因很可能是同時使用GPU不同進程之間的衝突。在這種情況下，我發現獲得張量流的最可靠的方法是重新啓動機器。在最壞的情況下，重新安裝tensorflow和/或NVidia驅動程序。當GPU突然停止工作時，請參閱one more case。