1.先上图列一下需要的安装环境
各个库版本号
cuda10.2.89+cudnn7.6.5
pytorch 1.8.1
conda install pytorch==1.8.1 cudatoolkit==10.2.89 -c pytorch
安装缓慢可参考 conda install pytorch 安装缓慢的解决办法 - 知乎
conda install cudnn==7.6.5
pip install torchvision==0.9.1
pip install onnx==1.8.0
pip install onnxruntime==1.6.0
下载tensorrt7.2.1
https://developer.nvidia.com/nvidia-tensorrt-7x-download
下载7.2.1for Linux and CUDA10.2
cd /home/wx/Downloads/TensorRT