Skip to content

Instantly share code, notes, and snippets.

@a1678991
Last active November 11, 2023 10:20
Show Gist options
  • Save a1678991/3821243ea24c05b940d71236625675e2 to your computer and use it in GitHub Desktop.
Save a1678991/3821243ea24c05b940d71236625675e2 to your computer and use it in GitHub Desktop.
stable-diffusion-webui python3.11 + CUDA12 + TensorRT + xformers TORCH_COMMAND
# TensorRT
export COMMANDLINE_ARGS="--xformers"
export TORCH_COMMAND="pip install -U pip wheel build setuptools pyproject.toml fonts ninja colored onnxruntime && \
pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
pip install -U --pre tensorrt --extra-index-url https://pypi.nvidia.com && \
pip install -U polygraphy onnx-graphsurgeon --extra-index-url https://pypi.ngc.nvidia.com && \
pip install -U 'httpx==0.24.*' protobuf==3.20.2"
@a1678991
Copy link
Author

事前にinstall.pyの中身を見て依存関係を把握し、処理をすべて消しておくこと

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment