как настроить cuDnn с theano на Windows 7 64 бит


Я установил Theano фреймворк и включил CUDA на моей машине, однако когда я "импортировал theano" в моей консоли python, я получил следующее сообщение:

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

Теперь, когда "CuDNN недоступен", я скачал cuDnn с сайта Nvidia. Я также обновил 'path' в среде и добавил 'optimizer_including=cudnn 'in'.с.конфигурационный файл txt.

Затем я попытался снова, но потерпел неудачу, с:

>>> import theano
Using gpu device 0: GeForce GTX 950 (CNMeM is disabled, CuDNN not available)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:Anaconda2libsite-packagestheano__init__.py", line 111, in <module>
    theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()
  File "C:Anaconda2libsite-packagestheanosandboxcudateststest_driver.py", line 31, in test_nvidia_driver1
    profile=False)
  File "C:Anaconda2libsite-packagestheanocompilefunction.py", line 320, in function
    output_keys=output_keys)
  File "C:Anaconda2libsite-packagestheanocompilepfunc.py", line 479, in pfunc
    output_keys=output_keys)
  File "C:Anaconda2libsite-packagestheanocompilefunction_module.py", line 1776, in orig_function
    output_keys=output_keys).create(
  File "C:Anaconda2libsite-packagestheanocompilefunction_module.py", line 1456, in __init__
    optimizer_profile = optimizer(fgraph)
  File "C:Anaconda2libsite-packagestheanogofopt.py", line 101, in __call__
    return self.optimize(fgraph)
  File "C:Anaconda2libsite-packagestheanogofopt.py", line 89, in optimize
    ret = self.apply(fgraph, *args, **kwargs)
  File "C:Anaconda2libsite-packagestheanogofopt.py", line 230, in apply
    sub_prof = optimizer.optimize(fgraph)
  File "C:Anaconda2libsite-packagestheanogofopt.py", line 89, in optimize
    ret = self.apply(fgraph, *args, **kwargs)
  File "C:Anaconda2libsite-packagestheanogofopt.py", line 230, in apply
    sub_prof = optimizer.optimize(fgraph)
  File "C:Anaconda2libsite-packagestheanogofopt.py", line 89, in optimize
    ret = self.apply(fgraph, *args, **kwargs)
  File "C:Anaconda2libsite-packagestheanosandboxcudadnn.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:

>>>

Кто-нибудь может мне помочь? Спасибо.

2 10

2 ответа:

Должен быть способ сделать это, установив только переменную окружения Path, но я никогда не мог заставить это работать. Единственное, что мне удалось, - это вручную скопировать файлы CuDNN в соответствующие папки в вашей установке CUDA.

Например, если ваша установка CUDA находится в C:\CUDA\v7.0 и вы извлекли CuDNN, чтобы C:\CuDNN вы бы скопировали следующим образом:

  • содержание C:\CuDNN\lib\x64\ будет скопировано в C:\CUDA\v7.0\lib\x64\
  • содержание из C:\CuDNN\include\ будет скопировано в C:\CUDA\v7.0\include\
  • содержание C:\CuDNN\bin\ будет скопировано в C:\CUDA\v7.0\bin\

После этого он должен работать.

В дополнение ко всем материалам, которые вы сделали, я обновил следующее содержание .с.txt в моей домашней папке, и после этого он работал.

[lib] 
#cnmem=1.0 
cudnn=1.0