tensorflow/keras调用CPU而不使用GPU问题

今天遇到一个bug，使用一个tensorflow后端的keras模型的时候，速度巨慢，后面发现是模型只使用了CPU，没有调用GPU。

1.解决方案

https://blog.csdn.net/qq_40829288/article/details/90509417
这个问题是由于tensorflow的版本高于tensorflow-gpu版本
首先查看版本

pip install

tensorflow 1.12.0
tensorflow-gpu 1.11.0
所以脚本默认调用了tensorflow而不是tensorflow-gpu
只需要将tensorflow-gpu升级即可

 pip install --upgrade tensorflow-gpu

注意！！！
如果直接使用上述命令，会将tensorflow-gpu升级到最新版本，注意和cuda版本的匹配
https://blog.csdn.net/qq_27825451/article/details/89082978

如何查看cuda版本

nvcc -V

所以升级时最好设置对应版本，例如

 pip install --upgrade tensorflow-gpu==1.12.0

2. 如何查看tensorflow只使用了CPU，没有使用GPU

python

进入python后
依次输入

 import tensorflow as tf
 sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

会打印出调用的是CPU还是GPU

3. 如何查看tensorflow能不能使用GPU

下面链接里面的方法也可以
https://blog.csdn.net/qq_40829288/article/details/90509417

也可以直接

python
import tensorflow as tf
tf.test.is_gpu_available()

便可以查看到信息