【问题标题】:ValueError «You are trying to use the old GPU back-end» when importing keras导入 keras 时出现 ValueError «您正在尝试使用旧的 GPU 后端»
【发布时间】:2017-11-29 15:03:45
【问题描述】:

我在 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 了解更多信息。

我该如何解决这个问题?

【问题讨论】:

  • 可能是环境变量THEANO_FLAGS。如果您将变量设置为包含device=cuda 而不是device=gpu,它将被修复。
  • 嗨@WilmarvanOmmeren,我现在刚刚尝试过,它奏效了。如果您将评论作为答案,我会将其标记为正确答案。谢谢。

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


【解决方案1】:

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

在 GPU 上工作时,还将浮点精度设置为 float32,因为这通常更快 (THEANO_FLAGS='device=cuda,floatX=float32')。

关于这个变量的更多信息可以在herehere找到。

【讨论】:

  • 导出 THEANO_FLAGS=mode=FAST_RUN,device=cuda,floatX=float32
【解决方案2】:

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

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

【讨论】:

  • 您能否详细说明一下这究竟是什么意思以及它如何回答这个问题?
猜你喜欢
  • 2018-01-01
  • 2018-04-14
  • 2019-10-09
  • 1970-01-01
  • 2020-11-30
  • 2020-09-24
  • 1970-01-01
  • 1970-01-01
  • 2023-03-14
相关资源
最近更新 更多