【发布时间】:2017-11-29 15:03:45
【问题描述】:
我在 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,它将被修复。 -
嗨@WilmarvanOmmeren,我现在刚刚尝试过,它奏效了。如果您将评论作为答案,我会将其标记为正确答案。谢谢。
标签: python machine-learning neural-network keras theano