我正在尝试使用Theano和gpu.我的操作系统是Ubuntu 16.04
首先,打字import theano会导致
Using cuDNN version 5110 on context None
Mapped name None to device cuda0: GeForce GTX 1080 (0000:01:00.0)
要查看我的GPU是否正在使用,请尝试使用theano文档进行测试
我的〜/ .theanorc是
[global]
device = cuda0
floatX = float32
[nvcc]
fastmath = True
在这种情况下测试说:
[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 …