2017-03-24 8 views
2

私はgpuでTheanoを使用しようとしています。私の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を入力して、まず

のUbuntu 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古いバックエンドを使用しては言う:

[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に置き換えようとします。デバイスCUDAに名前なしのマップされていないコンテキストなし にcuDNNバージョン5110を使用して

私はtheanoをインポートした後、同じ警告のようなテキストを持っていたのGeForce GT 750M(0000:01:00.0)

私は先に進み、DNNを訓練しました。スピードが私が前にCPUで実行したスピードよりはるかに速いことがわかります。だから、テキストがGPUが機能していないわけではないと思います。

関連する問題