2017-12-04 7 views
1
from tensorflow.python.client 
import device_lib 
def get_available_gpus(): 
local_device_protos = device_lib.list_local_devices() 
return [x.name for x in local_device_protos if x.device_type == 'GPU'] 
xx= get_available_gpus() 
print('The GPU device is: ', xx) 
print('Tensorflow: ', tf.__version__)` 

Dies gibt mir die folgende Ausgabe.Keras verwendet keine GPU auf Pycharm mit Python 3.5 und Tensorflow 1.4

Using TensorFlow backend. 
2017-12-04 18:13:37.479189: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA 
The GPU device is: [] 
Tensorflow: 1.4.0 

Ich benutze Pycharm Community Edition mit Python 3.5. Kann jemand helfen zu reparieren, wie man GPU anwenden kann. Obwohl ich beim Lesen der Dokumentation festgestellt habe, dass Keras GPU automatisch verwendet. Was ist das Problem in meinem Fall, dass es keine GPU verwendet? Ich habe auch versucht, Tensorflow zu deinstallieren, wie in einigen Fragen vorgeschlagen, aber es immer noch nicht verwendet. Brauchen Sie Hilfe.

Antwort

2

Ich löste es mit dem von Keras with TensorFlow backend not using GPU

Sie gegebene Antwort benötigen sowohl tensorflow und tensorflow-gpu zu deinstallieren. Und dann Tensorflow-GPU neu installieren. Wenn beide installiert sind, verwendet Keras nur die CPU-Version.