如何检查tensorflow gpu正在使用哪个cuda

【问题标题】：how to check which cuda is being used by tensorflow gpu如何检查tensorflow gpu正在使用哪个cuda
【发布时间】：2020-05-03 11:37:36
【问题描述】：

在我的笔记本上安装了cuda三个版本，分别是8.0、9.0和10.0，都是在path环境中配置的。当我使用tensorflow-gpu 2.0.0时，如何知道要部署哪个版本的cuda，而不考虑当前版本的tensorflow只兼容cuda 10.0。有没有办法在 python 控制台上打印信息？

【问题讨论】：

标签： tensorflow

【解决方案1】：

我在这里找到了答案get the CUDA and CUDNN version on windows with Anaconda installe：

from tensorflow.python.platform import build_info as tf_build_info
print(tf_build_info.cuda_version_number)
#10.0
print(tf_build_info.cudnn_version_number)
#7

【讨论】：