PyTorch 转 TensorRT：Python 和 C++ 端环境搭建

一、配置TensorRT

1. 解压安装文件：

cd ~/0-software/
tar xzvf TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz

2. 运行sudo gedit ~/.bashrc，在文档中添加：

export PATH=/home/wangbo/0-software/TensorRT-8.6.1.6/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/wangbo/0-software/TensorRT-8.6.1.6/lib

二、配置PyTorch

1. 新建conda环境

conda create -n base_113
conda activate base_113

2. 安装PyTorch 1.13.1

conda install pytorch==1.13.1 torchvision==0.14.1 pytorch-cuda=11.6 -c pytorch -c nvidia

3. 安装pycuda
   去whl下载网站，根据cuda和python确定版本号，例如：2022.1
   然后运行安装命令：

pip install cuda-python
pip install pycuda == 2022.1

4. 安装TensorRT

pip install tensorrt==8.6.1