jetson如何安装tensorrt
时间: 2023-11-20 15:04:59 浏览: 289
安装 TensorRT 可以参考以下步骤:
1. 下载 TensorRT 安装包,可以从 NVIDIA 官网下载。选择适合你的 Jetson 系统版本和 CUDA 版本的 TensorRT 安装包。
2. 解压安装包并进入解压后的目录,执行以下命令安装 TensorRT:
```
$ sudo dpkg -i nv-tensorrt-*.deb
$ sudo apt-get update
```
3. 安装完成后,设置环境变量:
```
$ export LD_LIBRARY_PATH=/usr/lib/aarch64-linux-gnu:$LD_LIBRARY_PATH
```
4. 确认安装是否成功:
```
$ dpkg -l | grep TensorRT
```
如果看到类似于以下输出,说明已经成功安装:
```
ii libnvinfer-dev 5.1.6-1+cuda10.2 arm64 TensorRT development libraries and headers
ii libnvinfer-plugin-dev 5.1.6-1+cuda10.2 arm64 TensorRT plugin libraries
ii libnvinfer-plugin5 5.1.6-1+cuda10.2 arm64 TensorRT plugin libraries
ii libnvinfer5 5.1.6-1+cuda10.2 arm64 TensorRT runtime libraries
ii libnvparsers-dev 5.1.6-1+cuda10.2 arm64 TensorRT parsers libraries and headers
ii libnvparsers5 5.1.6-1+cuda10.2 arm64 TensorRT parsers libraries
ii python-libnvinfer 5.1.6-1+cuda10.2 arm64 Python bindings for TensorRT
ii python-libnvinfer-dev 5.1.6-1+cuda10.2 arm64 Python development package for TensorRT bindings
ii python3-libnvinfer 5.1.6-1+cuda10.2 arm64 Python 3 bindings for TensorRT
ii python3-libnvinfer-dev 5.1.6-1+cuda10.2 arm64 Python 3 development package for TensorRT bindings
```
安装完成后,你就可以在 Jetson 上使用 TensorRT 进行深度学习推理任务了。
阅读全文