如何安装与已安装torch==1.8.0兼容的最新torch-tensorrt版本？

首先明确：torch==1.8.0兼容的最新torch-tensorrt版本是0.8.0，该版本适配torch 1.8.x系列，无需升级现有torch环境。

操作步骤： #

1. 清理已被升级的不兼容组件（若你的环境已自动升级到torch 2.0.1）：

pip uninstall -y torch torch-tensorrt

2. 重新安装指定版本的torch==1.8.0（根据你的CUDA版本选择对应包，以CUDA 11.1为例）：

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

如果是CUDA 10.2，替换为torch==1.8.0+cu102即可。
3. 安装兼容的torch-tensorrt 0.8.0，同时禁止自动升级依赖：

pip install torch-tensorrt==0.8.0 --no-deps

或者通过强制指定torch版本确保环境稳定：

pip install torch-tensorrt==0.8.0 torch==1.8.0 --force-reinstall

关键说明： #

• 直接执行pip install torch-tensorrt时，pip会默认安装最新版torch-tensorrt（当前为1.4.0），而该版本依赖torch 2.0+，因此会自动升级你的torch环境。
• 必须指定与torch 1.8.0匹配的torch-tensorrt版本，同时使用--no-deps避免pip自动更新依赖包。

内容的提问来源于stack exchange，提问作者Ars ML