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

在我的笔记本电脑中，安装了三个版本的

cuda

、8.0、9.0和10.0，所有这些都是在环境

path

中配置的。当我使用

tensorflow gpu 2.0.0

时，如何知道要部署哪个版本的

cuda

，而不考虑当前版本的tensorflow仅与cuda 10.0兼容。有没有办法在python控制台上打印信息？

我在这里找到了答案：

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