linux 源码编译安装TensorRT

~菜脯蛋~

已于 2023-07-18 10:10:50 修改

阅读量599

点赞数

分类专栏： linux 文章标签： linux 运维服务器

于 2023-03-30 22:15:13 首次发布

本文链接：https://blog.youkuaiyun.com/zhaozebin666/article/details/129867959

版权

linux 专栏收录该内容

7 篇文章

订阅专栏

文章详细介绍了如何在Linux系统上安装TensorRT，包括从NVIDIA官网下载压缩包，解压到指定目录，配置环境变量，使用pip安装Python所需的包（TensorRT,graphsurgeon,onnx_graphsurgeon），以及验证安装是否成功的过程。

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

linux 压缩包安装TensorRT

1. 下载压缩包

从NVIDIA官网下载TensorRT，需要注册登陆NVIDAI账号
例如：
在这里插入图片描述

2. 放到想要的目录下进行解压：

tar -zxvf TensorRT-8.6.0.12.Linux.x86_64-gnu.cuda-11.8.tar.gz
# 解压后会在当前目录下生成TensorRT-8.6.0.12文件夹

3. 配置环境变量

vim ~/.bashrc
# 添加以下内容
TensorRT_DIR=/home/zhaozb/TensorRT-8.6.0.12
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$TensorRT_DIR/lib
export PATH=$PATH:$TensorRT_DIR/bin
source ~/.bashrc

4. 安装python需要的包

cd /home/zhaozb/TensorRT-8.6.0.12/python
pip install tensorrt-8.2.5.1-cp38-none-linux_x86_64.whl

cd /home/zhaozb/TensorRT-8.6.0.12/graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl

cd /home/zhaozb/TensorRT-8.6.0.12/onnx_graphsurgeon
pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl

5. 验证

python3
>>>import tensorrt
>>>print(tensorrt.__version__)
>>>assert tensorrt.Builder(tensorrt.Logger())