나는 GPU에 keras를 사용 autoencoder_layers.py 실행하려고하지만 난이 오류

Question

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 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' 

.