我在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。
如何解决这个问题?
回答:
您应该更改(或添加)名为THEANO_FLAGS
的环境变量。如果您将变量设置为包含device=cuda
而不是device=gpu
,错误将会消失。
在使用GPU时,也请将浮点精度设置为float32
,因为这通常会快得多(THEANO_FLAGS='device=cuda,floatX=float32'
)。