如何在Linux中查看CUDA及cuDNN版本?

2023-12-26 11阅读

答案: 在Linux系统下,查看CUDA及其依赖库的版本是非常重要的。而对于深度学习开发者来说,cuDNN也是必不可少的一个库。本文将介绍在Linux系统下如何查看CUDA和cuDNN版本。

如何在Linux中查看CUDA及cuDNN版本?

CUDA 版本查询

1. 查看 CUDA 的安装路径

第一我们需要知道 CUDA 的安装路径,在终端输入以下命令:

```

which nvcc

如果输出结果为 /usr/local/cuda/bin/nvcc,则说明 CUDA 已经被正确地安装了,并且默认安装路径为 /usr/local/cuda/。

2. 查看 CUDA 版本

接着可以使用以下命令来查看当前机器上已经安装好的 CUDA 的版本号:

nvcc --version

这时会输出一些信息,其中包括 NVIDIA 安装驱动程序、Toolkit 和 cuDNN 所需的各个组件和工具包的详细信息以及 CUDA 版本号等。例如:

从上面截图中可以得到当前机器上已经成功地安装了 NVIDIA 驱动程序(Driver Version)、CUDA Toolkit(release) 以及 cuDNN 所需的各个组件和工具包(CUDA Runtime Version)。

cuDNN 版本查询

1. 查看 cuDNN 的安装路径

我们需要知道 cuDNN 的安装路径。在终端输入以下命令:

ls /usr/local/cuda/include/

如果输出结果中含有 cudnn.h,则说明已经正确地将 cuDNN 安装到了 CUDA 路径下。

2. 查看 cuDNN 版本号

接着可以使用以下命令来查看当前机器上已经安装好的 cuDNN 的版本号:

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

从截图中可得出该机器上已经成功地安装了 cuDNN v7.6.5。

文章版权声明:除非注明,否则均为游侠云资讯原创文章,转载或复制请以超链接形式并注明出处。

目录[+]