代码之家 › 专栏 › 技术社区 › user288609

关于模块的tensorflow。python站台build_info没有属性cudnn_version_number

cudnn tensorflow2.0 tensorflow

0

user288609 · 技术社区 · 4 年前

我试图从tensorflow打印出cudnn信息,如下所示

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

然而,它的输出是,这个错误的原因是什么?

AttributeError                            Traceback (most recent call last)
<ipython-input-2-439c29a4890a> in <module>
----> 1 print(tf_build_info.cudnn_version_number)

AttributeError: module 'tensorflow.python.platform.build_info' has no attribute 'cudnn_version_number'

0 回复 | 直到 4 年前

1

0

mrtpk 3 年前

从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 是一种有序的信息技术。因此,要获得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)

2

0

Ad5001 3 年前

使用python 3.6、tf 2.5 cuda 11.2 cudnn 8.2。带有Geforce MX150卡的win10笔记本电脑:

from tensorflow.python.platform import build_info as tf_build_info
print("cudnn_version",tf_build_info.build_info['cudnn_version'])
 > cudnn_version 64_8 
print("cuda_version",tf_build_info.build_info['cuda_version'])
 > cuda_version 64_112