From 7dd6fc9f8ab3f9052454e97c0953f7435e62125f Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 16 Mar 2023 01:59:53 -0400 Subject: [PATCH] Put a command that actually works for installing xformers for nvidia. --- README.md | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index b5a0c48f..277d5d99 100644 --- a/README.md +++ b/README.md @@ -61,13 +61,9 @@ AMD users can install rocm and pytorch with pip if you don't have it already ins ### NVIDIA -Nvidia users should install torch using this command: +Nvidia users should install torch and xformers using this command: -```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118``` - -Nvidia users should also install Xformers for a speed boost but can still run the software without it. - -```pip install xformers``` +```pip install torch==1.13.1 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 xformers``` #### Troubleshooting