与GPU一起使用时,Theano导入错误

时间:2016-12-01 13:20:06

标签: python theano theano-cuda

我安装了带有numpy 1.11.2和scipy 0.18.1版本的Anaconda Python 2.7.12。我通过conda安装安装了Theano 0.8.2。我在.theanorc.txt文件中添加了这些行:

[global]
floatX = float32 
device = gpu0

[nvcc]
compiler_bindir=C:\Program Files (x86)\Microsoft Visual Studio 12.0\VC\bin`

我在环境变量中设置了CUDA路径。我还在cuda安装中添加了CUDNN v4.0。

然而,当我做一个简单的

import theano

在python解释器中它给了我以下错误并回退到cpu。

`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/<USER_NAME>/AppData/Local/Theano/compiledir_Windows-10-10.0.14393-Intel64_Family_6_Model_94_Stepping_3_GenuineIntel-2.7.12-64/cuda_ndarray/cuda_ndarray.lib and object C:/Users/<USER_NAME>/AppData/Local/Theano/compiledir_Windows-10-10.0.14393-Intel64_Family_6_Model_94_Stepping_3_GenuineIntel-2.7.12-64/cuda_ndarray/cuda_ndarray.exp

ERROR (theano.sandbox.cuda): Failed to compile cuda_ndarray.cu: DLL load failed: The specified module could not be found.
WARNING (theano.sandbox.cuda): CUDA is installed, but device gpu0 is not available (error: cuda unavailable)`

错误未指定缺少哪个DLL。我不确定这是bug还是theano安装/配置问题或CUDA安装/配置问题。

由于

1 个答案:

答案 0 :(得分:0)

在重新安装CUDAv8.0之后工作,很奇怪,因为特定于GPU的CUDA样本就像devicequery一样正常工作..