使用 Anaconda install 在 Windows 上获取 CUDA 和 CUDNN 版本

Posted 2023-02-23

【中文标题】使用 Anaconda install 在 Windows 上获取 CUDA 和 CUDNN 版本

【英文标题】：get the CUDA and CUDNN version on windows with Anaconda installe

【发布时间】：2019-02-23 21:37:56

【问题描述】：

在Windows上使用Anaconda安装了一个tensorflow-gpu版本，如何查看它的CUDA和CUDNN版本？谢谢。

【问题讨论】：

使用 conda list 它会显示你的包版本

【参考方案1】：

虽然不是公开记录的 API，但您目前可以像这样访问它：

from tensorflow.python.platform import build_info as tf_build_info
print(tf_build_info.cuda_version_number)
# 9.0 in v1.10.0
print(tf_build_info.cudnn_version_number)
# 7 in v1.10.0

【讨论】：

@iyop45 感谢您的反馈，您是对的，最近的 TensorFlow 版本不再在 build_info 中包含该信息。我有opened an issue about it，这似乎是弃用 CMake for Windows 构建以支持 Bazel 的结果。

@iyop45 这刚刚在this commit 上得到修复。但不确定哪个版本会首先提供该修复。

【参考方案2】：

您也可以从 anaconda 命令行运行 conda list：

conda list cudnn

# packages in environment at C:\Anaconda2:
#
# Name                    Version                   Build  Channel
cudnn                     6.0                           0

【参考方案3】：

使用以下命令检查 Conda 安装的 CUDA：

conda list cudatoolkit

以及以下命令检查 conda 安装的 CUDNN 版本：

conda list cudnn

如果您想通过 CONDA 安装/更新 CUDA 和 CUDNN，请使用以下命令：

conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn

您也可以使用以下命令检查 CUDA 安装：

nvidia-smi

或

nvcc --version

【讨论】：

nvidia-smi 没有给你安装的版本，只给你支持的版本，这个问题没用，

nvcc --version 如果你安装了 cuda 工具包和 conda，则在 anaconda 提示符下不起作用。

请注意，在 conda 中，如果要为 pytorch 或 tensorflow 安装 cudatoolkit 和 cudnn，则不应手动安装。使用其中任何一个自动覆盖依赖项的 conda 安装程序。

【参考方案4】：

从 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，因此在未来的版本中可能会有所改变。在以前的版本中，我们可以像在 jdehesa 的 answer 中那样做 from tensorflow.python.platform import build_info as tf_build_info; print(tf_build_info.cuda_version_number)。