torch-tensorrt 安装报错：如何正确使用 pip 进行安装？

安装 torch-tensorrt 报错？

在尝试使用 pip 安装 torch-tensorrt 时，遇到了以下错误：

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.

这是因为 pypi 上发布的 torch-tensorrt 只是一个占位符项目，实际的安装需要使用以下命令：

pip install torch-tensorrt -f https://github.com/nvidia/torch-tensorrt/releases

然而，即使指定了 -f 参数，pip 仍然尝试从默认的仓库（如清华源）下载 torch-tensorrt，导致了错误。

解决方法：

将 torch-tensorrt 的 wheel 文件下载到本地，然后使用以下命令进行安装：

pip install torch-tensorrt-<version>-py3-none-any.whl

其中是 torch-tensorrt 的版本号。

以上就是torch-tensorrt 安装报错：如何正确使用 pip 进行安装？的详细内容，更多请关注硕下网其它相关文章！