how to setup cuDnn with theano on Windows 7 64 bit

问题:

I have installed Theano framework and enabled CUDA on my machine, however when I "import theano" in my python console, I got the following message:

>>> import theano Using gpu device 0: GeForce GTX 950 (CNMeM is disabled, CuDNN not available) 

Now that "CuDNN not available", I downloaded cuDnn from Nvidia website. I also updated 'path' in environment, and added 'optimizer_including=cudnn' in '.theanorc.txt' config file.

Then, I tried again, but failed, with:

>>> import theano Using gpu device 0: GeForce GTX 950 (CNMeM is disabled, CuDNN not available) Traceback (most recent call last):   File "", line 1, in    File "C:\Anaconda2\lib\site-packages\theano\__init__.py", line 111, in      theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()   File "C:\Anaconda2\lib\site-packages\theano\sandbox\cuda\tests\test_driver.py", line 31, in test_nvidia_driver1     profile=False)   File "C:\Anaconda2\lib\site-packages\theano\compile\function.py", line 320, in function     output_keys=output_keys)   File "C:\Anaconda2\lib\site-packages\theano\compile\pfunc.py", line 479, in pfunc     output_keys=output_keys)   File "C:\Anaconda2\lib\site-packages\theano\compile\function_module.py", line 1776, in orig_function     output_keys=output_keys).create(   File "C:\Anaconda2\lib\site-packages\theano\compile\function_module.py", line 1456, in __init__     optimizer_profile = optimizer(fgraph)   File "C:\Anaconda2\lib\site-packages\theano\gof\opt.py", line 101, in __call__     return self.optimize(fgraph)   File "C:\Anaconda2\lib\site-packages\theano\gof\opt.py", line 89, in optimize     ret = self.apply(fgraph, *args, **kwargs)   File "C:\Anaconda2\lib\site-packages\theano\gof\opt.py", line 230, in apply     sub_prof = optimizer.optimize(fgraph)   File "C:\Anaconda2\lib\site-packages\theano\gof\opt.py", line 89, in optimize     ret = self.apply(fgraph, *args, **kwargs)   File "C:\Anaconda2\lib\site-packages\theano\gof\opt.py", line 230, in apply     sub_prof = optimizer.optimize(fgraph)   File "C:\Anaconda2\lib\site-packages\theano\gof\opt.py", line 89, in optimize     ret = self.apply(fgraph, *args, **kwargs)   File "C:\Anaconda2\lib\site-packages\theano\sandbox\cuda\dnn.py", line 2508, in apply     dnn_available.msg) AssertionError: cuDNN optimization was enabled, but Theano was not able to use it. We got this error: Theano can not compile with cuDNN. We got this error:  >>> 

anyone can help me? Thanks.

回答1:

There should be a way to do it by setting only the Path environment variable but I could never get that to work. The only thing that worked for me was to manually copy the CuDNN files into the appropriate folders in your CUDA installation.

For example, if your CUDA installation is in C:\CUDA\v7.0 and you extracted CuDNN to C:\CuDNN you would copy as follows:

• The contents of C:\CuDNN\lib\x64\ would be copied to C:\CUDA\v7.0\lib\x64\
• The contents of C:\CuDNN\include\ would be copied to C:\CUDA\v7.0\include\
• The contents of C:\CuDNN\bin\ would be copied to C:\CUDA\v7.0\bin\

After that it should work.

回答2:

In addition to all the stuffs you did I updated following content of .theanorc.txt in my home folder and it worked after that.

[lib]  #cnmem=1.0  cudnn=1.0 

文章来源: how to setup cuDnn with theano on Windows 7 64 bit