mirror of https://github.com/bmaltais/kohya_ss
17 lines
482 B
Plaintext
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
|