From 80665081e0ee3a12921429a3171018ae28c85703 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Mon, 13 Mar 2023 13:50:54 -0400 Subject: [PATCH] Add command to install unstable pytorch builds for ROCm. --- README.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index b49d0256..5c80b10a 100644 --- a/README.md +++ b/README.md @@ -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. ### 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``` + +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 users should install torch using this command: