2017-12-04 6 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__)` 

これは私に次の出力を与えます。Python 3.5とTensorflow 1.4を持つPycharmでKerasはGPUを使用していません

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 

私はpython 3.5でPycharmコミュニティ版を使用しています。 誰でもGPUの使用方法を修正する手助けをすることができます。ドキュメントを読んでいる間、私はKerasがGPUを自動的に使用することを発見しました。私の場合、GPUを使用しないという問題は何ですか? 私はTensorflowをアンインストールしようとしましたが、いくつかの質問で示唆されているように、まだそれを使用していません。ガイダンスが必要です。

答えて

2

私はあなたがtensorflowとtensorflow-GPUの両方をアンインストールする必要がKeras with TensorFlow backend not using GPU

によって与えられた答えでそれを解決しました。そして、tensorflow-gpuを再インストールするよりも。 両方がインストールされている場合、KerasはCPUバージョンのみを使用します。

関連する問題