我正在尝试使用gpu上的keras运行autoencoder_layers.py，但是我会遇到此错误

Question

autoencoder_layers.py github代码

import theano
from keras import backend as K
from keras.backend.theano_backend import _on_gpu
from keras.layers.convolutional import Convolution2D, UpSampling2D
from keras.layers.core import Dense, Layer
from theano import tensor as T
from theano.sandbox.cuda import dnn

但是我得到了这个错误：

 /home/hoda/anaconda2/bin/python /home/hoda/Downloads/keras-convautoencoder-master/autoencoder_layers.py
Using gpu device 0: GeForce GTX 970M (CNMeM is disabled, cuDNN not available)
Using Theano backend.
Traceback (most recent call last):
  File "/home/hoda/Downloads/keras-convautoencoder-master/autoencoder_layers.py", line 3, in <module>
    from keras.backend.theano_backend import _on_gpu
ImportError: cannot import name _on_gpu

我该如何修复？

Answer 1

评论行from keras.backend.theano_backend import _on_gpu并将_on_gpu定义为：

def _on_gpu():
    return theano.config.device[:3] == 'gpu'

它将解决您的错误。