关于在 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 文件中更改它们。

关于在 GPU 上使用 Theano 设置 Keras

regarding setting up Keras with Theano on GPU

theano

keras