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

使用Anaconda installe在windows上获取CUDA和CUDNN版本

cudnn gpu anaconda tensorflow python

2

user297850 · 技术社区 · 6 年前

2 回复 | 直到 4 年前

1

12

John H 6 年前

你也可以跑 conda list

conda list cudnn

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

2

20

sidk 5 年前

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

conda list cudatoolkit

conda list cudnn

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

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

或者,您可以使用以下命令检查CUDA安装:

nvidia-smi

nvcc --version

3

9

javidcf 6 年前

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

4

0

mrtpk 4 年前

从TensorFlow 2.4.1开始,我们可以使用 tensorflow.python.platform.build_info 为了获取关于哪个CUDA的信息,cuDNN二进制文件是根据哪个CUDA构建的。

>>> 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) answer .