我在Ubuntu 16.04上使用Keras和Theano后端。我的设置一直没有问题,但是,当我导入Keras(import keras
)时突然出现以下错误:
ValueError:您正在尝试使用旧的GPU后端。它是从Theano中删除的。现在使用device = cuda *。有关详细信息,请参阅https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29。
如何解决此问题?
答案 0 :(得分:17)
您应该更改(或添加)名为THEANO_FLAGS
的环境变量。如果您将变量设置为包含device=cuda
而不是device=gpu
,则错误将消失。
可以找到有关此变量的更多信息here。
答案 1 :(得分:-1)
对我来说没有gpu,请使用cpu,工作:
export THEANO_FLAGS='mode=FAST_RUN,device=cpu,floatX=float32'