298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 10:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update install instructions for torch.

Browse Source
This commit is contained in:
comfyanonymous 2023-03-15 01:58:27 -04:00
parent ed4810e670
commit 54593db6dc
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@ -56,19 +56,14 @@ At the time of writing this pytorch has issues with python versions higher than
### AMD (Linux only) ### AMD (Linux only)
AMD users can install rocm and pytorch with pip if you don't have it already installed, this is the command to install the stable version: AMD users can install rocm and pytorch with pip if you don't have it already installed, this is the command to install the stable version:
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2``` ```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2```
I highly recommend you use the nightly/unstable pytorch builds though because they work a lot better for me (run this in the ComfyUI folder so it picks up the requirements.txt):
```pip install --upgrade --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/rocm5.4.2 -r requirements.txt```
### NVIDIA ### NVIDIA
Nvidia users should install torch using this command: Nvidia users should install torch using this command:
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117``` ```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. Nvidia users should also install Xformers for a speed boost but can still run the software without it.