关于在GPU上与Theano建立Keras

Question

我一直在关注setting up keras on GPU的说明。这就是我按方法3所做的那样。

import theano
theano.config.device = 'gpu'
theano.config.floatX = 'float32'

但是，我收到以下错误消息

Using Theano backend.
Traceback (most recent call last):
File "train.py", line 17, in 
theano.config.device = 'gpu'
File "/devl/geophys/dplearn/tensorflow/tf_0.12/lib/python3.4/site-     
packages/theano/configparser.py", line 338, in set

"Can't change the value of this config parameter "

Exception: Can't change the value of this config parameter after initialization!

Answer 1

问题在于，当您使用Theano和GPU时 - 在加载包时，它会加载并编译一些非pythonic函数（例如来自CUDA）。这就是为什么更改此选项会导致异常 - 在预编译代码中无法更改此选项的原因。要设置此设置，您需要在路径THEANO_FLAGS或.theanorc文件中更改它们。