コンピュータには、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アルゴリズムです)
何が問題になりますか?