当前位置: 首页 > news >正文

在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_MAJORCUDNN_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_MAJORNV_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 。

相关文章:

  • Nat Commun | 单细胞和空间转录组学分析构建NSCLC免疫和非免疫区图谱
  • 示波器直流耦合和交流耦合的区别
  • MySQL为什么默认使用RR隔离级别?
  • 24. 状态模式
  • 栈/堆/static/虚表
  • Pytest基础使用
  • 【产品小白】如何运营一个成熟的产品
  • 《深度剖析:BERT与GPT——自然语言处理架构的璀璨双星》
  • SeaTunnel简易封装方案:让数据集成更简单
  • 「低延迟+快速集成:Amazon IVS如何重塑实时互动视频体验?」
  • python中的min函数的key的用法 - abs绝对值
  • 误杀!微软道歉了!
  • 【CVPR2024-工业异常检测】PromptAD方法(CLIP和提示学习)
  • HTML5响应式使用css媒体查询
  • 基于灵动微单片机SPIN系列的两轮车解决方案
  • 【记录】使用 Docker 搭建 MongoDB 分布
  • windows 平台编译openssl
  • Web开发-JS应用原生代码前端数据加密CryptoJS库jsencrypt库代码混淆
  • unityAB包(1/2)
  • Arduino示例代码讲解:Multiple tone player 多音播放器
  • 盐城经济技术开发区党工委书记王旭东接受纪律审查和监察调查
  • 大环线呼之欲出,“金三角”跑起来了
  • 2000多年前的“新衣”长这样!马王堆文物研究新成果上新
  • 联合国报告:全球经济前景恶化,面临高度不确定性
  • 选址江南制造总局旧址,上海工业博物馆建设有新进展
  • 昆明警方重拳打击经济领域违法犯罪:去年抓获905名嫌犯