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

Question

我已通过pip安装了tensorflow-gpu 1.12，我想知道这个二进制文件是使用哪个CUDA和cuDNN版本构建的。是否有一种方式可以直接从tensorflow软件包中查找此信息？例如：

import tensorflow as tf
tf.get_build_information()

我尝试过在谷歌、stackoverflow和tensorflow文档中搜索，但没有找到有用的信息。

我认为它可能存在，因为numpy实现了类似的功能：

import numpy as np
np.show_config()

- Ian Riley

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

注意：由于这不是公共API，未来版本可能会发生变化。在之前的版本中，我们可以执行以下操作：

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

- mrtpk