Linux 和 Windows 查看 CUDA 和 cuDNN 版本

Linux

查看 CUDA 版本

  • 方法一:
nvcc --version

nvcc -V

如果 nvcc 没有安装,那么用方法二。

  • 方法二:

去安装目录下查看:

cat /usr/local/cuda/version.txt

查看 cuDNN 版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果没有,那么可能没有安装 cuDNN。

如果是为了使用 PyTorch/TensorFlow,在 Linux 服务器上推荐使用 conda 安装,使用 conda 可以很方便安装 PyTorch/TensorFlow 以及对应版本的 CUDA 和 cuDNN。

Windows

查看 CUDA 版本

在命令行中执行:

nvcc --version

或者进入 CUDA 的安装目录查看:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

查看 cuDNN 版本

进入 CUDA 的安装目录查看文件 cudnn.h :(注意修改v9.0)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h

如下所示,cuDNN 版本为 7.2.1 :

如果不知道安装路径,或者安装了多个版本的 CUDA,可以去环境变量内查看 CUDA_PATH 或 path。

使用 PyTorch 查看 CUDA 和 cuDNN 版本

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

References

ubuntu: 查看cuda版本 -- JNingWei
Windows查看CUDA版本 -- 胡大炮的妖孽人生
LINUX下查询CUDA和CUDNN版本 -- 许文_北京北
Linux和Windows下查看环境变量方法 -- 小寒

作者:wuliytTaotao
本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接。
posted @ 2019-09-03 16:32  wuliytTaotao  阅读(150087)  评论(1编辑  收藏  举报