2017-02-26 7 views
4

WindowsのKerano & Theanoの実行インストールがあります(これはtutorialに従う)。今私はバックエンドをTensorflowに切り替えようとしましたが、これはかなりうまくいきました。Windows上でTensorFlow 1.0にGPUが表示されません(ただしTheanoはあります)

私が持っている唯一の問題は、対照的にTheanoが行うことTensorflow does not detect my GPU、次のとおりです。

from tensorflow.python.client import device_lib 
def get_available_gpus(): 
    local_device_protos = device_lib.list_local_devices() 
    return [x.name for x in local_device_protos if x.device_type == 'GPU'] 

は結果が得られていないが、Theanoバックエンドで実行している場合、それはかなりうまく動作します:どうやら

C:\Programming\Anaconda3\python.exe D:/cnn_classify_cifar10.py 
Using Theano backend. 
DEBUG: nvcc STDOUT nvcc warning : The 'compute_20', 'sm_20', and 'sm_21' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning). 
mod.cu 
    Creating library C:/Users/Alex/AppData/Local/Theano/compiledir_Windows-10-10.0.14393-SP0-Intel64_Family_6_Model_60_Stepping_3_GenuineIntel-3.5.2-64/tmpgsy496fe/m91973e5c136ea49268a916ff971b7377.lib and object C:/Users/Alex/AppData/Local/Theano/compiledir_Windows-10-10.0.14393-SP0-Intel64_Family_6_Model_60_Stepping_3_GenuineIntel-3.5.2-64/tmpgsy496fe/m91973e5c136ea49268a916ff971b7377.exp 

Using gpu device 0: GeForce GTX 770 (CNMeM is enabled with initial size: 80.0% of memory, cuDNN 5005) 

いくつかの構成がありませんが、私は何がわかりません。 Theanoを正しく実行するには、私は、ファイルには、以下の内容で~/.theanorcと呼ばれる必要:

[global] 
device = gpu 
floatX = float32 

[cuda] 
root = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 

[nvcc] 
flags=-LC:C:\Programming\WinPython-64bit-3.5.2.2\python-3.5.2.amd64\libs 

はたぶん似た何かが欠落しているか、多分私はadd environment variables like for Theano?する必要があります。おそらくrelated question Linux(?)です。

Gistには、完全なインストールログ(奇妙な例外が含まれています)があります。

TensorflowにGPUを表示させる方法はありますか?

+0

ためtensorflow(CPUのみ)または(のみGPU用)tensorflow-GPUのどちらかをインストールしますか? – Steven

+0

pipインストールのテンソルフローを使用してホイールに直接URLを指定して、my [gist file](https://gist.github.com/apacha/a595c244f90a27aced56f67f7598d90d)を参照してください。 –

+0

コマンドが間違っています。あなたはpip3をインストールする必要があります - アップグレードテンソルフロー - gpu – Steven

答えて

4

同じマシンにtensorflowとtensorflow-gpuの両方をインストールすると、現時点で問題が発生することがあります。

あなたがtensorflowをインストールしましたか、バージョン1.0

+0

ハァッ!この答えを見つけるには、グーグルグーグルが多くなりました。彼らは、 'pyd'ファイルを何の苦情もなく上書きするようです。なぜこの小さな事実は文書化されていないのですか? tensorflow-gpuはテンソルフローに加えて代用品ではないと考えています。 – dashesy

+0

私は彼らがなぜそれを作ったのかについての良い答えはありません。そのため、現時点でドキュメントをインストールしたり、ドキュメンテーションが貧弱な理由をインストールすることはできません。私は(何かについて)ドキュメンテーションの前にプルリクエストを試みましたが、実際の機能が新しいバージョンで変更されるため、数か月後に私はただクローズしました。 – Steven

関連する問題