ubuntu下tensorrt安装

TensorRT是英伟达推出的一个高性能深度学习推理库，可用于优化训练后的深度学习模型以在NVIDIA GPU上进行加速推理。在Ubuntu下安装TensorRT的步骤如下：

1. 下载TensorRT安装包并解压缩，可以通过官网下载，链接为：https://developer.nvidia.com/nvidia-tensorrt-download
2. 进入解压缩后的文件夹，运行以下命令：

$ cd TensorRT-<version>/python
$ sudo pip3 install tensorrt-<version>.whl
$ sudo pip3 install graphsurgeon-<version>.whl

其中，`<version>`为TensorRT的版本号。
3. 设置环境变量，运行以下命令：

$ echo "export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64:${LD_LIBRARY_PATH}" >> ~/.bashrc
$ source ~/.bashrc

其中，`<version>`为CUDA的版本号。

至此，在Ubuntu下安装TensorRT完成。