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

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

 pi