流扩散安装失败

Question

我尝试过的命令：

python -m streamdiffusion.tools.install-tensorrt

上述命令生成的错误：

Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com, https://pypi.nvidia.com
Collecting tensorrt-bindings
  Downloading tensorrt-bindings-9.2.0.post12.dev5.tar.gz (6.8 kB)
  Preparing metadata (setup.py) ... error
  error: subprocess-exited-with-error

  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [16 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "C:\Users\Owner\AppData\Local\Temp\pip-install-b73tmy1x\tensorrt-bindings_53f09f435d4542748e4e8641602be581\setup.py", line 137, in <module>
          raise RuntimeError(open("ERROR.txt", "r").read())
      RuntimeError:
      ###########################################################################################
      The package you are trying to install is only a placeholder project on PyPI.org repository.
      This package is hosted on NVIDIA Python Package Index.
     
      This package can be installed as:
      ```
      $ pip install --no-cache-dir --extra-index-url https://pypi.nvidia.com tensorrt-bindings
      ```
      ###########################################################################################
     
      [end of output]

  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.

我在没有 venv 虚拟环境的情况下执行此操作。 我尝试使用 TensorRT 更快地生成图像，但出现此错误。 还有一个约定俗成的代号叫

pipe.enable_xformers_memory_efficient_attention()

。 使用这个，代码没有什么大问题。 但是，它会减慢生成速度。

有人可以告诉我如何解决这个错误吗？

Answer 1

错误说明了一切：

      The package you are trying to install is only a placeholder project on PyPI.org repository.
      This package is hosted on NVIDIA Python Package Index.
     
      This package can be installed as:
      ```
      $ pip install --no-cache-dir --extra-index-url https://pypi.nvidia.com tensorrt-bindings
      ```

同样，正如@journpy建议的那样，请检查https://pypi.org/project/tensorrt/以获取有关安装

TensorRT

模块的详细信息。

你必须先执行：

pip install --no-cache-dir --extra-index-url https://pypi.nvidia.com tensorrt-bindings

然后重新启动其余模块的安装过程（如果有的话）。