安装 Torch-TensorRT 时为什么会出现“占位符项目”错误,如何解决?

安装 torch-tensorrt 时为什么会出现“占位符项目”错误,如何解决?

pip 安装 torch-tensorrt 报错原因

在使用 pip 安装 torch-tensorrt 时,可能会遇到以下错误,提示不适用于 pypi 的占位符项目:

runtimeerror:
the package you are trying to install is only a placeholder project on pypi.org repository.
to install torch-tensorrt please run the following command:
$ pip install torch-tensorrt -f https://github.com/nvidia/torch-tensorrt/releases

错误根源

此错误源自上传到 pypi 的 torch-tensorrt 仅是一个占位符项目。这是因为真实包在 github 存储库上提供。

解决方案

要正确安装 torch-tensorrt,需要直接从 github 下载轮文件(.whl):

  1. 前往 github 存储库:https://github.com/nvidia/torch-tensorrt/releases
  2. 找到与你的系统相对应的 .whl 文件(例如,带有 .cu116 后缀的文件表示 cuda 11.6)。
  3. 手动将 .whl 文件下载到本地。
  4. 在终端中,使用以下命令安装本地 .whl 文件:
pip install /path/to/downloaded.whl

注意:

  • 请确保已根据系统要求安装必要的 cuda 和 cudnn 版本。
  • 如果仍遇到问题,请尝试使用 --user 标志安装 torch-tensorrt,这将将其安装到本地用户空间而不是系统范围内。

以上就是安装 Torch-TensorRT 时为什么会出现“占位符项目”错误,如何解决?的详细内容,更多请关注硕下网其它相关文章!