运行deepmind时出错

Question

我花了两天时间来安装deepQ(python版本)的要求，然后我今天试着运行它，但我遇到了这个问题，代码如下。

root@unicorn:/media/trump/Data1/wei/college/laboratory/deep_q_rl-master/deep_q_rl# python run_nips.py
A.L.E: Arcade Learning Environment (version 0.5.0)
[Powered by Stella]
Use -help for help screen.
Warning: couldn't load settings file: ./ale.cfg
Game console created:
ROM file:  ../roms/breakout.bin
Cart Name: Breakout - Breakaway IV (1978) (Atari)
Cart MD5:  f34f08e5eb96e500e851a80be3277a56
Display Format:  AUTO-DETECT ==> NTSC
ROM Size:        2048
Bankswitch Type: AUTO-DETECT ==> 2K

Running ROM file...
Random seed is 65
Traceback (most recent call last):
File "run_nips.py", line 60, in <module>
launcher.launch(sys.argv[1:], Defaults, __doc__)
File "/media/trump/Data1/wei/college/laboratory/deep_q_rl-master/deep_q_rl/launcher.py", line 223, in launch
rng)
File "/media/trump/Data1/wei/college/laboratory/deep_q_rl-master/deep_q_rl/q_network.py", line 53, in __init__
num_actions, num_frames, batch_size)
File "/media/trump/Data1/wei/college/laboratory/deep_q_rl-master/deep_q_rl/q_network.py", line 168, in build_network
batch_size)
File "/media/trump/Data1/wei/college/laboratory/deep_q_rl-master/deep_q_rl/q_network.py", line 407, in build_nips_network_dnn
from lasagne.layers import dnn
File "/usr/local/lib/python2.7/dist-packages/Lasagne-0.2.dev1-py2.7.egg/lasagne/layers/dnn.py", line 13, in <module>
raise ImportError("dnn not available")  # pragma: no cover
ImportError: dnn not available

我已经测试过theano，numpy，scipy，没有出现错误。但是当我运行它的时候，它显示dnn不可用。所以我来找dnn，代码是这样的

import theano
from theano.sandbox.cuda import dnn

from .. import init
from .. import nonlinearities
from .base import Layer

from .conv import conv_output_length
from .pool import pool_output_length
from ..utils import as_tuple

if not theano.config.device.startswith("gpu") or not dnn.dnn_available():
    raise ImportError("dnn not available")  # pragma: no cover

只希望有人能帮我。

Answer 1

你安装CUDA和cuDNN了吗？千层面是建立在Theano之上的，在某些情况下，它依赖于cuda代码(例如这里)，而不是将其抽象出来。

这可以从导入中看到：

from theano.sandbox.cuda import dnn

另请参阅：https://github.com/Lasagne/Lasagne/issues/242

要获得cuDNN，您需要在NVidia上注册为开发人员，请参阅：https://developer.nvidia.com/accelerated-computing

希望这能有所帮助。

干杯，迈克尔