2017-02-26 1 views
4

Windows에서 Kerano & Theano를 실행 중입니다 (다음은 tutorial에 따라). 이제 백엔드를 Tensorflow으로 전환하려고 시도했습니다.TensorFlow 1.0은 Windows에서 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) 

누락 된 구성이 있지만 무엇이 있는지 모르겠습니다. 누락되거나 어쩌면 내가 add environment variables like for Theano? 할 필요가있다 유사한

[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 

아마 뭔가 : Theano 제대로 실행하기 위해, 나는 다음과 같은 내용으로 ~/.theanorc라는 파일을 필요로했다. Linux (?)에서 related question 일 수 있습니다.

전체 설치 로그 (이상한 예외 포함)는 Gist에서 찾을 수 있습니다.

Tensorflow에서 볼 수있는 GPU 만드는 방법은 어떤 아이디어입니까?

+0

에 대한 (전용 GPU의 경우) 또는 tensorflow-GPU (CPU 전용) 중 하나 tensorflow를 설치 하시겠습니까? – Steven

+0

pip를 사용하여 tensorflow를 설치하고 직접 url을 사용하여 my [gist file] (https://gist.github.com/apacha/a595c244f90a27aced56f67f7598d90d)를 참조하십시오. –

+0

명령이 잘못되었습니다. 당신 pip3 설치 - 업 그레 이드 텐서 흐름 - GP를 – Steven

답변

4

동일한 기계에 tensorflow와 tensorflow-gpu를 모두 설치하면 문제가 발생할 수 있습니다.

는 tensorflow를 설치 했는가 버전 1.0

+0

허! 이 답변을 찾으려면 인터넷 검색이 더 많이 필요했습니다. 그들은 어떤 불평없이'pyd' 파일을 덮어 쓴 것처럼 보입니다. 왜이 작은 사실이 문서화되어 있지 않습니까? Tenorflow-gpu는 tensorflow에 추가 된 것이지 대체물이 아니라고 생각합니다. – dashesy

+0

나는 그들이 왜 그렇게 만들었는지에 대한 좋은 대답을하지 못했습니다. 그래서 지금은 그저 설치하거나 문서가 왜 부족한 가를 설치할 수 없습니다. 나는 (실제로는) 문서에 대한 풀 요청을 시도했지만 실제로 기능이 이후 버전에서 변경 될 것이기 때문에 몇 달 후에 그냥 닫았다. – Steven