0

在尝试在 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 中知道我的设备是什么，即“gpu”、“gpu1”还是“gpu0”？我已经为我的案例尝试了所有 3 个，但没有产生结果。

任何建议将不胜感激。

4

1 回答 1

1

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

import os
os.environ['THEANO_FLAGS'] = "device=cuda,force_device=True,floatX=float32"
import theano
于 2017-06-02T14:55:51.503 回答