ubuntu 安装tensorrt

最新推荐文章于 2025-07-18 10:43:19 发布

oyjwin

最新推荐文章于 2025-07-18 10:43:19 发布

阅读量303

点赞数 1

CC 4.0 BY-SA版权

文章标签： python ubuntu

本文介绍如何安装TensorRT 8.0版本，适用于CUDA 11.2和cuDNN 8.2环境。文章详细说明了下载、解压、配置环境变量及Python环境的过程，并提供了验证安装是否成功的步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、查看cuda、cudnn版本

操作链接：链接

二、选择对应的tensorrt版本，这里选择tar.gz格式的文件，推荐使用；deb安装出现问题；

我的cuda版本11.2；cudnn版本8.2，选择tensorrt 8.0版本下载安装

如需下载7版本，链接为：

二、将压缩包解压

tar -xzvf TensorRT-8.0.0.3.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz

cd TensorRT-8.0.0.3

配置环境

export LD_LIBRARY_PATH=/home/project/package/tensort/TensorRT-8.0.0.3/lib:$LD_LIBRARY_PATH

激活环境

source ~/.bashrc

三、打开python env

cd TensorRT-8.0.0.3/python

选择对应版本安装

python3.6 对应 cp36

pip install tensorrt-8.0.0.3-cp36-none-linux_x86_64.whl

验证安装

import tensorrt

tensorrt.__version__

安装成功；

为了后续模型转换方便，还可以安装torch2trt包

torch2trt：
https://github.com/NVIDIA-AI-IOT/torch2trt

安装参考指令：

git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
sudo python setup.py install --plugins