如何在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。