2017-06-20 6 views
1

私はnvidia-smiのグラフィックスカードを見ることができるPythonがインストールされたLinuxマシンを使用しています。virtualenvのテンソルがGPUデバイスを見つけられない

私はtensorflow内のローカルデバイスを一覧表示、

python3 -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())" 

私が正しく二つのデバイス、CPUとGPUを参照してください。リストのデバイスにコマンドを私はvirtualenvのを起動し、私はそれにtensorflowインストール、

virtualenv -p /usr/bin/Python3 ~/.virtualenvs/test 
source ~/.virtualenvs/test/bin/activate 
pip install tensorflow 

場合

しかし、

python3 -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())" 

は出力GPUデバイスをしません。ここには何が欠けていますか?

私はCUDAライブラリがtensorflowによって見つからないと思っていますが、修正方法がわかりません。

答えて

1

gpuを使用するには、正しいpythonパッケージはではなくtensorflow-gpuです。

関連する問題