ubuntu安装下载tensorrt
时间: 2023-10-26 16:08:27 浏览: 110
下载和安装TensorRT在Ubuntu系统上需要执行以下步骤:
1. 解压并配置环境变量:
```
# 解压
tar -zxvf TensorRT-8.2.5.1.Linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz
# 配置环境变量
sudo gedit ~/.bashrc # 使用vim也可以
# 在文件末尾添加以下两条路径,根据解压实际路径进行修改
export LD_LIBRARY_PATH=$PATH:/home/liujinfu/Downloads/TensorRT-8.2.5.1/lib:$LD_LIBRARY_PATH
export LIBRARY_PATH=$PATH:/home/liujinfu/Downloads/TensorRT-8.2.5.1/lib::$LIBRARY_PATH
# 保存后刷新环境变量
source ~/.bashrc
```
2. 安装TensorRT库:
```
cd TensorRT-8.2.5.1/python # 进入python目录
# 根据Python版本安装,假设为python3.9
pip install tensorrt-8.2.5.1-cp39-none-linux_x86_64.whl
# 安装依赖
cd TensorRT-8.2.5.1/graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl
```
3. 安装依赖:
```
pip install 'pycuda<2021.1' # 注意onnxruntime-gpu的版本需与实际环境匹配,这里选择1.11版本
pip install onnxruntime-gpu==1.11
```
阅读全文