在Orin上查看CUDA cuDNN TensorRT的版本
目录
一、CUDA版本
方法1:使用 nvcc 命令
方法2:查看 CUDA 安装目录下的版本文件
方法3:使用 nvidia-smi 命令
二、cuDNN版本
方法1:查看 cuDNN 头文件信息
方法2:使用 dpkg 命令(适用于通过 Debian 包安装的情况)
方法3:查看库文件的软链接信息
三、TensorRT版本
方法1:使用 Python 脚本查看
方法2:查看 TensorRT 头文件
方法3:使用 dpkg 命令(适用于通过 Debian 包安装的情况)
方法4:直接查找libnvinfer.so
一、CUDA版本
方法1:使用 nvcc
命令
nvcc
是 CUDA 的编译器驱动程序,通过查看其版本信息可以间接得知 CUDA 的版本。
打开终端,输入以下命令:
nvcc --version
或者nvcc -V
执行该命令后,会输出类似如下的信息:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Sun_Oct_23_22:16:07_PDT_2022
Cuda compilation tools, release 11.4, V11.4.315
Build cuda_11.4.r11.4/compiler.31964100_0
其中 release 11.4
就表明当前系统中安装的 CUDA 版本是 11.4。
方法2:查看 CUDA 安装目录下的版本文件
CUDA 安装后,会在特定目录下存在一个版本文件,从中可以获取版本信息。一般来说,CUDA 安装在 /usr/local/cuda
目录下,使用以下命令查看版本文件内容:
cat /usr/local/cuda/version.txt
输出结果可能如下:
CUDA Version 11.4.35
这显示当前 CUDA 版本为 11.4。
有可能不存在/usr/local/cuda/version.txt,但是有/usr/local/cuda/version.json文件,也可以看到CUDA版本。
方法3:使用 nvidia-smi
命令
nvidia-smi
是 NVIDIA 系统管理接口,它可以提供有关 NVIDIA GPU 设备的详细信息,其中也包含 CUDA 的版本信息。
在终端中输入以下命令:
nvidia-smi
输出结果中会有类似如下的内容:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 520.61.05 Driver Version: 520.61.05 CUDA Version: 11.8 |
|-------------------------------+----------------------+----------------------+
这里的 CUDA Version: 11.8
表明当前系统支持的 CUDA 版本为 11.8。需要注意的是,nvidia-smi
显示的 CUDA 版本是 GPU 驱动程序支持的最高 CUDA 版本,实际安装的 CUDA 版本可能小于或等于这个值。
二、cuDNN版本
可以通过以下几种方法查看当前安装的 cuDNN 版本:
方法1:查看 cuDNN 头文件信息
cuDNN 的版本信息通常存储在其头文件中,可以通过查看该头文件内容来获取版本号。cuDNN 头文件一般位于 /usr/include
目录下,文件名是 cudnn_version.h
或 cudnn.h
。
打开终端,使用以下命令查看版本信息:
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
或者如果文件名为 cudnn.h
,则使用:
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
输出结果示例如下:
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
/* cannot use constexpr here since this is a C-only file */
cuDNN 的完整版本号是通过 CUDNN_MAJOR
、CUDNN_MINOR
和 CUDNN_PATCHLEVEL
组合而成的。在上述示例中,版本号为 8.6.0 。计算方式为 CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL
,即 8 * 1000 + 6 * 100 + 0 = 8600
,对应版本 8.6.0 。
方法2:使用 dpkg
命令(适用于通过 Debian 包安装的情况)
如果是通过 Debian 包(.deb 文件)安装的 cuDNN,那么可以使用 dpkg
命令来查看安装包的版本信息。
在终端中执行以下命令:
dpkg -l | grep libcudnn
输出结果可能如下:
ii libcudnn8 8.6.0.166-1+cuda11.4 arm64 cuDNN runtime libraries
ii libcudnn8-dev 8.6.0.166-1+cuda11.4 arm64 cuDNN development libraries and headers
ii libcudnn8-samples 8.6.0.166-1+cuda11.4 arm64 cuDNN samples
从输出中可以看到 cuDNN 的版本号是 8.6.0 。
方法3:查看库文件的软链接信息
cuDNN 库文件通常会有软链接指向具体版本的库文件,通过查看软链接信息也能获取版本号。
使用以下命令查看库文件的软链接:
ls -l /usr/lib/aarch64-linux-gnu/libcudnn.so
或者
ls -l /usr/lib/aarch64-linux-gnu/libcudnn.so*
输出结果示例:
rwxrwxrwx 1 root root 13 May 16 2023 /usr/lib/aarch64-linux-gnu/libcudnn.so -> libcudnn.so.8
lrwxrwxrwx 1 root root 17 Sep 22 2022 /usr/lib/aarch64-linux-gnu/libcudnn.so.8 -> libcudnn.so.8.6.0
-rw-r--r-- 1 root root 153408 Sep 22 2022 /usr/lib/aarch64-linux-gnu/libcudnn.so.8.6.0
从这里可以直接看出 cuDNN 的版本是 8.6.0 。
三、TensorRT版本
可以通过以下几种方法查看当前安装的 TensorRT 版本:
方法1:使用 Python 脚本查看
如果已经安装了 TensorRT 的 Python 绑定,那么可以通过编写一个简单的 Python 脚本来查看版本信息。
打开终端,创建一个 Python 文件(例如 check_trt_version.py
),并在其中添加以下代码:
import tensorrt as trt
print(f"TensorRT版本: {trt.__version__}")
然后在终端中运行该脚本:
python3 check_trt_version.py
运行脚本后,会输出当前安装的 TensorRT 版本号。
方法2:查看 TensorRT 头文件
TensorRT 的版本信息通常存储在其头文件中,可以通过查看头文件内容来获取版本号。头文件一般位于 /usr/include
目录下,文件名为 NvInferVersion.h
。
在终端中使用以下命令查看版本信息:
grep "#define NV_TENSORRT_MAJOR" /usr/include/NvInferVersion.h
grep "#define NV_TENSORRT_MINOR" /usr/include/NvInferVersion.h
grep "#define NV_TENSORRT_PATCH" /usr/include/NvInferVersion.h
如果NvInferVersion.h不在/usr/include目录下,可以先sudo find / -iname NvInferVersion.h找到它。
输出结果示例如下:
#define NV_TENSORRT_MAJOR 8 //!< TensorRT major version.
#define NV_TENSORRT_MINOR 5 //!< TensorRT minor version.
#define NV_TENSORRT_PATCH 2 //!< TensorRT patch version.
完整的 TensorRT 版本号是通过 NV_TENSORRT_MAJOR
、NV_TENSORRT_MINOR
和 NV_TENSORRT_PATCH
组合而成的。在上述示例中,版本号为 8.5.2 。
方法3:使用 dpkg
命令(适用于通过 Debian 包安装的情况)
如果是通过 Debian 包(.deb 文件)安装的 TensorRT,那么可以使用 dpkg
命令来查看安装包的版本信息。
在终端中执行以下命令:
dpkg -l | grep tensorrt
输出结果可能如下:
ii tensorrt 8.5.2.2-1+cuda11.4 arm64 Meta package for TensorRT
从输出中可以看到 TensorRT 的版本号是 8.5.2 。
方法4:直接查找libnvinfer.so
从输出中可以看到 TensorRT 的版本号是 8.5.2 。