TensorflowではGPUが表示されず、代わりにCPUを使用します。

Question

GPU対応のTensorflowを使用してPythonスクリプトを実行しています。しかし、プログラムはGPUを認識していないようで、すぐにCPUを使い始める。これの原因は何でしょうか？

Answer 1

はただ、他の言葉で、tensorflowはGPUを検出するが、その上に任意のOPを派遣することはできません、原因CUDA initialization failureにGPUを見を停止することがをtensorflow議論に追加するので、CPUにフォールバックします。このケースでは、原因が同時にGPUを使用して、異なるプロセス間の競合である可能性が高い。この

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

のように、ログにエラーが表示されるはずです。その場合、テンソルフローを得るために見つけた最も信頼できる方法は、マシンを再起動することです。最悪の場合、テンソルフローやNVidiaドライバを再インストールしてください。

GPUが突然動作を停止するとone more caseも参照してください。