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

2024-04-20 02:47:48 发布

您现在位置:Python中文网/ 问答频道 /正文


Tags: python
3条回答

您还可以从anaconda命令行运行conda list

conda list cudnn

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

尽管不是公共文档化的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

使用以下命令通过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

相关问题 更多 >