web-dev-qa-db-ja.com

Windows 764ビットでtheanoを使用してcuDnnをセットアップする方法

Theanoフレームワークをインストールし、マシンでCUDAを有効にしましたが、pythonコンソールでtheanoをインポート」すると、次のメッセージが表示されます。

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

「CuDNNが利用できません」ということで、NvidiaのWebサイトからcuDnnをダウンロードしました。また、環境内の「パス」を更新し、「。theanorc.txt」構成ファイルに「optimizer_include = cudnn」を追加しました。

次に、再試行しましたが、失敗しました。

>>> 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:\Anaconda2\lib\site-packages\theano\__init__.py", line 111, in <module>
    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:

>>>

誰かが私を助けることができますか？ありがとう。

pythontheanocudnn
10
Tyler Xie

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 \にコピーされます

その後、それは動作するはずです。

35
Alexander Measure