Ubuntu安装tensorrt
时间: 2023-09-22 17:06:34 浏览: 83
以下是在 Ubuntu 上安装 TensorRT 的步骤:
1. 首先需要从 NVIDIA 官网下载 TensorRT,下载地址为:https://developer.nvidia.com/tensorrt
2. 解压 TensorRT 安装包:
```
tar xzvf <TensorRT-*.tar.gz>
```
3. 进入 TensorRT 目录并安装 TensorRT:
```
cd TensorRT-*
sudo ./install.sh
```
4. 安装完成后,需要添加 TensorRT 到系统环境变量中,可以通过在 /etc/profile.d/ 目录下创建一个新的脚本文件来实现:
```
sudo nano /etc/profile.d/tensorrt.sh
```
然后在文件中添加以下内容:
```
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/TensorRT/lib
export PATH=$PATH:/usr/local/TensorRT/bin
```
5. 保存并退出文件,然后重新加载环境变量:
```
source /etc/profile.d/tensorrt.sh
```
6. 最后,可以通过运行以下命令来验证 TensorRT 是否已经正确安装:
```
nvcc -V
```
如果输出类似于以下内容,则说明 TensorRT 已经安装成功:
```
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Nov_30_19:10:16_PST_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0
```
希望对你有所帮助!
阅读全文