TensorFlow查看GPU是否可用

tensorflow查看GPU是否可用

import tensorflow as tf
tf.test.is_gpu_available()

返回为True表示可以使用，返回为False表示不可以使用

note：

• 不可使用的话需要注意查看自己的tensorflow是不是下载的tensorflow-gpu，如果安装的是普通的tensorflow，是不能在GPU上进行加速的。
• FALSE还有一个原因可能就是，tensorflow的版本与cuda、cudnn、python版本没有相对应，需要注意一定得对应起来，对应关系可以参考官方给出的文档。点击跳转
• CUDA(也就是cudatoolkit)版本虽然是可以向下兼容的，但是还是建议和官网一致。

---

在一次的项目中，由于项目tensorflow版本为1.14.0，所以按照官网的要求就务必安装CUDA 10 ，CUDNN 7.4，但是CUDNN 7.4就没办法安装到虚拟环境里面，所以本人安装了CUDNN为7.6.0的版本。然后测试发现也是可以的，故我个人认为，安装好CUDA之后，安装CUDNN只需要通过conda search cudnn --info找到CUDA支持的CUDNN就可以了。

😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃😃

如果有帮助的话，留个赞再走吧 ,感谢 😃，有任何问题评论区留言，看到我会和大家一起探索的。