kohya_ss/requirements_linux_rocm.txt

17 lines
482 B
Plaintext

# Custom index URL for specific packages
--extra-index-url https://download.pytorch.org/whl/rocm6.3
--find-links https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1
torch==2.7.1+rocm6.3
torchvision==0.22.1+rocm6.3
tensorboard==2.14.1; python_version=='3.11'
tensorboard==2.16.2; python_version!='3.11'
tensorflow-rocm==2.14.0.600; python_version=='3.11'
tensorflow-rocm==2.16.2; python_version!='3.11'
# no support for python 3.11
onnxruntime-rocm==1.21.0
-r requirements.txt