Question

尝试在GPU（安装了CUDA）上运行我的Keras代码时，我无法执行以下语句，正如许多在线参考文献所建议的那样。

set THEANO_FLAGS="mode=FAST_RUN,device=gpu,floatX=float32" & python theanogpu_example.py

我收到以下错误。

ValueError: Invalid value ("FAST_RUN,device=gpu,floatX=float32") for configurati
on variable "mode". Valid options are ('Mode', 'DebugMode', 'FAST_RUN', 'NanGuar
dMode', 'FAST_COMPILE', 'DEBUG_MODE')

我也尝试过从代码中建议的其他模式。

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

我收到以下错误。

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

除了知道如何让它运行外，我还想借此机会提出一个更简单的问题。如何在Windows中了解我的设备是什么，即＆＃39; gpu＆＃39;或者＆＃39; gpu1＆＃39;或＆＃39; gpu0＆＃39;？我已经为我的情况尝试了所有3但是它没有产生结果。

任何建议都将受到赞赏。

Answer 1

最好的方法是在运行代码之前使用THEANO_FLAGS，因为导入Theano后无法更改配置变量，请尝试：

import os
os.environ['THEANO_FLAGS'] = "device=cuda,force_device=True,floatX=float32"
import theano

Keras与Theano在GPU上

1 个答案: