python - 如何在 tensorflow 中获取现有 tensorflow 二进制文件的构建信息？

Question

我已经通过 pip 安装了 tensorflow-gpu 1.12。我想知道这个二进制文件是针对哪个 CUDA、cuDNN 构建的。有没有办法直接从 tensorflow 包中找到这些信息？例如

import tensorflow as tf
tf.get_build_information()

我试过搜索 google、SO 和 tensorflow 文档，但没有找到任何有用的东西。

我认为它可能存在，因为 numpy 实现了类似的东西:

import numpy as np
np.show_config()

Answer 1

从 TensorFlow 2.4.1 开始，我们可以使用 tensorflow.python.platform.build_info要获取有关构建二进制文件的 CUDA、cuDNN 的信息。

>>> import tensorflow
>>> print(tensorflow.__version__)
'2.4.1'
>>> import tensorflow.python.platform.build_info as build
>>> print(build.build_info)
OrderedDict([('cpu_compiler', '/usr/bin/gcc-5'), ('cuda_compute_capabilities', ['sm_35', 'sm_50', 'sm_60', 'sm_70', 'sm_75', 'compute_80']), ('cuda_version', '11.0'), ('cudnn_version', '8'), ('is_cuda_build', True), ('is_rocm_build', False)])

build.build_info是一个 OrderedDict。所以要获得 CuDNN 和 CUDA 版本:

>>> print(build.build_info['cuda_version'])
11.0
>>> print(build.build_info['cudnn_version'])
8

注意:由于这不是公共(public) API，因此在 future 版本中可能会发生变化。在以前的版本中，我们可以这样做:

from tensorflow.python.platform import build_info as tf_build_info;
print(tf_build_info.cuda_version_number)