ValueError«您正在尝试使用旧的GPU后端»导入keras时

时间:2017-06-26 17:35:16

标签: python machine-learning neural-network keras theano

我在Ubuntu 16.04上使用KerasTheano后端。我的设置一直没有问题,但是,当我导入Keras(import keras)时突然出现以下错误:

  

ValueError:您正在尝试使用旧的GPU后端。它是从Theano中删除的。现在使用device = cuda *。有关详细信息,请参阅https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29

如何解决此问题?

2 个答案:

答案 0 :(得分:17)

您应该更改(或添加)名为THEANO_FLAGS的环境变量。如果您将变量设置为包含device=cuda而不是device=gpu,则错误将消失。

可以找到有关此变量的更多信息here

答案 1 :(得分:-1)

对我来说没有gpu,请使用cpu,工作:

export THEANO_FLAGS='mode=FAST_RUN,device=cpu,floatX=float32'