InvokeAI/environments-and-requirements/requirements-lin-amd.txt
Lincoln Stein 3a1724652e
upgrade requirements to CUDA 11.7, torch 1.13 (#2331)
* upgrade requirements to CUDA 11.7, torch 1.13

* fix ROCm version number

Co-authored-by: Lincoln Stein <lstein@gmail.com>
2023-01-16 14:19:27 -05:00

7 lines
242 B
Plaintext

-r environments-and-requirements/requirements-base.txt
# Get hardware-appropriate torch/torchvision
--extra-index-url https://download.pytorch.org/whl/rocm5.2 --trusted-host https://download.pytorch.org
torch>=1.13.1
torchvision>=0.14.1
-e .