如何解决使用 pip 安装 torch-tensorrt 时出现的 "error: subprocess-exited-with-error" 错误?

如何解决使用 pip 安装 torch-tensorrt 时出现的

pip 安装 torch-tensorrt 报错

问题:

使用 pip 安装 torch-tensorrt 时,出现了以下错误:

error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.

解释:

这个错误的含义是,pip 在尝试从 github 下载并安装 torch-tensorrt 时失败了。这是因为上传到 pypi 的 torch-tensorrt 只是一个占位项目,无法直接安装。

虽然 pip 命令中指定了 -f 选项,这表示先从指定仓库(例如 github)下载,但 pip 仍然先去清华源等默认仓库中搜索,导致下载失败。

解决方案:

  • github 手动下载 wheel 文件:

    访问 nvidia 的 torch-tensorrt github 仓库,下载适用于你的操作系统和 python 版本的 wheel 文件(以 .whl 结尾)。

  • 本地安装 wheel 文件:

    使用 pip 安装下载的 wheel 文件,例如:

    pip install torch-tensorrt-0.0.0.post1-py3-none-any.whl

完成这些步骤后,你应该可以成功安装 torch-tensorrt。

以上就是如何解决使用 pip 安装 torch-tensorrt 时出现的 "error: subprocess-exited-with-error" 错误?的详细内容,更多请关注硕下网其它相关文章!