Ubuntu18.04 conda环境中配置tensorrt,并安装torch2trt，如何将pytorch模型转换成tensorrt

创建conda环境

1. conda create --name tensorrt python=3.7
2. conda activate tensorrt
3. 官网下载TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.0.cudnn7.6.tar.gz（注意要先下载tensorrt，根据tensorrt的版本来设置cuda和cudnn的版本）
4. pip install pytorch==1.6.0 torchvision==0.7.0
5. cuda和cudnn的安装：conda可以在conda的虚拟环境里安装10.0以上的cudatoolkit和cudnn，但实验发现tensorrt会调用系统中的cuda和cudnn。所以需要在系统环境中安装对应版本的cuda和cudnn。

tar方法安装tensorrt

7. tar xzvf TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.0.cudnn7.6.tar.gz
8. vim ~/.bashrc ，添加环境变量，如果使用了zsh，要在 ~/.zshrc中添加该环境变量

export LD_LIBRARY_PATH=/home/ps/Downloads/TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.0.cudnn7.6/TensorRT-7.0.0.11/lib:${LD_LIBRARY_PATH}

9. cd TensorRT-7.0.0.11/python
pip install tensorrt-7