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__)`
这给了我以下输出。
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
我正在使用Pycharm社区版和python 3.5。 任何人都可以帮助修复如何强制使用GPU。虽然在阅读文档时,我发现Keras会自动使用GPU。那么在我的情况下,它不使用GPU的问题是什么? 我也试图卸载Tensorflow,如某些问题所示,但它仍然没有使用它。需要指导。
答案 0 :(得分:2)
我用Keras with TensorFlow backend not using GPU
给出的答案解决了这个问题您需要卸载tensorflow和tensorflow-gpu。而不是重新安装tensorflow-gpu。 如果两者都安装,Keras将只使用CPU版本。