使用Anaconda在Windows上安装了一个tensorflow-gpu版本,如何检查它的CUDA和CUDNN版本?谢谢。
答案 0 :(得分:4)
您还可以从anaconda命令行运行conda list
:
conda list cudnn
# packages in environment at C:\Anaconda2:
#
# Name Version Build Channel
cudnn 6.0 0
答案 1 :(得分: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
OR
nvcc --version
答案 2 :(得分: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
答案 3 :(得分: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,因此在未来版本中可能会发生变化。在以前的版本中,我们可以像在 jdehesa 的 answer 中那样做 from tensorflow.python.platform import build_info as tf_build_info; print(tf_build_info.cuda_version_number)
。