Commit 80665081 authored by comfyanonymous's avatar comfyanonymous
Browse files

Add command to install unstable pytorch builds for ROCm.

parent c6720009
...@@ -50,10 +50,16 @@ Put your VAE in: models/vae ...@@ -50,10 +50,16 @@ Put your VAE in: models/vae
At the time of writing this pytorch has issues with python versions higher than 3.10 so make sure your python/pip versions are 3.10. At the time of writing this pytorch has issues with python versions higher than 3.10 so make sure your python/pip versions are 3.10.
### AMD (Linux only) ### AMD (Linux only)
AMD users can install rocm and pytorch with pip if you don't have it already installed: 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.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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment