2017-03-24 5 views
2

나는 Theano with gpu를 사용하려고합니다. 내 GPU가 사용되는 경우 내 OS 내가 내 ~/.theanorc가Theano with gpu. 문맥에 cuDNN 버전 5110 사용 없음

[global] 
device = cuda0 
floatX = float32 

[nvcc] 
fastmath = True 
입니다 test from theano documentation

시도 보려면

Using cuDNN version 5110 on context None 
Mapped name None to device cuda0: GeForce GTX 1080 (0000:01:00.0) 

가 발생합니다 import theano를 입력 첫째

우분투 16.04입니다

이 경우 테스트에서는 다음과 같이 말합니다.

[GpuElemwise{exp,no_inplace}(<GpuArrayType<None>(float32, (False,))>), 
HostFromGpu(gpuarray)(GpuElemwise{exp,no_inplace}.0)] 
Looping 1000 times took 0.191431 seconds 
Result is [ 1.23178029 1.61879349 1.52278066 ..., 2.20771813 2.29967761 1.62323296] 
Used the cpu 
device = gpu0와 6,

그러나 기존 백엔드를 사용하여 말한다 :

[GpuElemwise{exp,no_inplace}(<CudaNdarrayType(float32, vector)>), HostFromGpu(GpuElemwise{exp,no_inplace}.0)] 
Looping 1000 times took 0.199280 seconds 
Result is [ 1.23178029 1.61879349 1.52278066 ..., 2.20771813 2.29967761 1.62323296] 
Used the gpu 

그래서 내가 뭔가 CUDA와 잘못 생각합니다. 괜찮은지 어떻게 확인할 수 있습니까? "컨텍스트"가 "없음"인 이유는 무엇입니까? 왜 테스트는 "cpu 사용"이라고 말합니까?

답변

4

cuda0을 cuda로 바꿔 봅니다.

내가 theano 가져온 후에 같은 경고와 같은 텍스트 가졌다 : 장치 CUDA에 이름 없음 매핑없는 상황에 맞는 없음 에 cuDNN 버전 5110을 사용하여

을 : 지포스 GT 750M (0000 : 01 : 00.0)

을 나는 앞서 가서 DNN을 훈련 시켰고, 이전에 CPU에서 코드를 실행 한 것보다 속도가 훨씬 빠르다는 것을 알 수있었습니다. 그래서, 텍스트가 GPU가 작동하지 않는다는 것을 의미하지는 않습니다.