使用pip安装TensorRT失败，报metadata-generation-failed错误求助

问题背景 #

使用Python 3.8.1rc1执行以下命令安装TensorRT时，出现metadata-generation-failed错误：

pip install nvidia-tensorrt
pip install torch-tensorrt

错误信息：

note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.

解决方案 #

1. 替换Python正式版本 #

Python 3.8.1rc1是候选发布版本，存在未修复的兼容性问题，很多PyPI包对RC版本的支持不完善。建议更换为Python 3.8系列的正式稳定版（如3.8.10），再尝试安装。

2. 升级pip及依赖工具 #

旧版本的pip、setuptools或wheel可能导致元数据生成失败，先执行升级：

pip install --upgrade pip setuptools wheel

3. 针对nvidia-tensorrt的安装优化 #

• 匹配CUDA版本：TensorRT与CUDA版本强绑定，需确认本地CUDA版本后选择对应TensorRT版本。例如CUDA 11.8对应TensorRT 8.6.x，CUDA 12.1对应TensorRT 9.x。
• 指定版本安装：避免自动拉取最新版本，手动指定兼容版本：

  pip install nvidia-tensorrt==8.6.1
  

• 本地wheel安装：若PyPI源安装失败，可从NVIDIA官方下载对应Python版本、CUDA版本的TensorRT wheel包，本地执行安装：

  pip install ./nvidia_tensorrt-8.6.1-cp38-none-linux_x86_64.whl
  

4. 针对torch-tensorrt的安装优化 #

• 匹配PyTorch与TensorRT版本：torch-tensorrt的版本需与已安装的PyTorch、TensorRT版本完全兼容。例如PyTorch 2.0+对应torch-tensorrt 1.4.x，且依赖TensorRT 8.6.x。
• 指定CUDA索引源安装：安装时指定对应CUDA版本的PyTorch源，确保依赖匹配：

  pip install torch-tensorrt --index-url https://download.pytorch.org/whl/cu118
  

  （将cu118替换为本地CUDA版本，如cu117、cu121）

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