我试图运行在GPU上使用keras autoencoder_layers.py,但我得到这个错误(I am trying to run autoencoder_layers.py using keras on gpu but i get this error)
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我该如何解决它?
autoencoder_layers.py github code
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 dnnbut I get this error:
/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_gpuhow can I fix it?
最满意答案
from keras.backend.theano_backend import _on_gpu注释行from keras.backend.theano_backend import _on_gpu定义为:
def _on_gpu(): return theano.config.device[:3] == 'gpu'它会解决你的错误。
Comment the line from keras.backend.theano_backend import _on_gpu and define _on_gpu as:
def _on_gpu(): return theano.config.device[:3] == 'gpu'It will fix your error.
更多推荐
发布评论