安装tensorrt.tar
时间: 2025-01-14 20:57:35 浏览: 35
### 如何安装 TensorRT Tar 文件
#### 下载 TensorRT 软件包
为了成功安装 TensorRT,需先从 NVIDIA 官方网站下载适用于目标操作系统的 tar 格式的软件包[^1]。
#### 解压文件
使用命令行工具解压缩已下载的 tar 文件。对于特定版本如 TensorRT 7.1.3.4,在 Ubuntu 16.04 上针对 CUDA 11.0 和 cuDNN 8.0 的组合,执行如下指令:
```shell
tar -xvzf TensorRT-7.1.3.4.Ubuntu-16.04.x86_64-gnu.cuda-11.0.cudnn8.0.tar.gz
```
#### 配置环境变量
编辑 `~/.bashrc` 文件来设置必要的环境变量以便于后续调用 TensorRT 库。具体来说就是向该配置文件追加两行导出语句指向 TensorRT 的 bin 及 lib 目录位置,并使更改生效:
```shell
gedit ~/.bashrc
# 在文件末尾添加以下内容
export PATH=/PATH/TO/TensorRT-7.1.3.4/bin:${PATH}
export LD_LIBRARY_PATH=/PATH/TO/TensorRT-7.1.3.4/lib:${LD_LIBRARY_PATH}
source ~/.bashrc
```
请注意 `/PATH/TO/TensorRT-7.1.3.4` 是占位符,实际应替换为具体的解压目录路径。
#### Python 绑定安装
进入 TensorRT 提供的 Python 接口所在的位置并通过 pip 工具完成 wheel 包形式分发的 Python 扩展模块安装工作。假设当前使用的 Python 版本号为 3.7,则对应的命令将是这样的:
```shell
cd /PATH/TO/TensorRT-7.1.3.4/python
pip install tensorrt-7.1.3.4-cp37-none-linux_x86_64.whl
```
阅读全文
相关推荐














