2017-05-06 37 views
2

コンピュータには、Anaconda(3)、TensorFlow(GPU)、Kerasというソフトウェアがインストールされています。 2つのAnaconda仮想環境があります.1つはTensorFlow for Python 2.7と3.5 for 1、どちらもGPUバージョンinstalled according to the TF instructionsです。 (私はTensorFlowのCPUバージョンが別々の環境で以前にインストールされていたが、私はそれを削除した。)私は、次の実行tensorflow-gpuがインストールされていても、GPUは計算に使用されません

source activate tensorflow-gpu-3.5 
python code.py 

、それが唯一の3MiB GPUのメモリ使用量を示しnvidia-smiをチェックしますPythonでは、GPUは計算に使用されていないようです。 (code.pyはKerasで実装された単純な深いQ-learningアルゴリズムです)

何が問題になりますか?

答えて

0

を私のGPUが実行されていなかった理由が原因CuDNNの壊れたインストールであった、より正確にはライブラリとソースがCuDNNの異なるバージョンから来ました。

the following piece of advice.

3

これらの問題をデバッグするには、どの操作がどのデバイスに割り当てられているかを確認するのが良い方法です。

あなたは、セッションに設定パラメータを渡すことでこれを確認することができます

session = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 

アプリを実行すると、デバイスが使用されているかを示すいくつかの出力が表示されます。

あなたはここでより多くの情報を見つけることができます。 https://www.tensorflow.org/tutorials/using_gpu

関連する問題