关于模块“tensorflow.python.platform.build_info”没有属性“cudnn_version_number”
关于模块“tensorflow.python.platform.build_info”没有属性“cudnn_version_number”
提问于 2021-01-31 16:01:20
我尝试从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'Stack Overflow用户
发布于 2021-04-28 01:42:21
从CUDA2.4.1开始，我们可以使用tensorflow.python.platform.build_info来获取有关构建二进制文件所基于的TensorFlow，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)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/65976909
复制相关文章
相似问题
腾讯云开发者
