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

Tsh*_*dau 13 python machine-learning neural-network theano keras

我在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.

我该如何解决这个问题?

Wil*_*ren 18

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

有关此变量的更多信息,请点击此处.

  • export THEANO_FLAGS = mode = FAST_RUN,device = cuda,floatX = float32 (4认同)