kohya_ss/requirements_linux_rocm.txt

5 lines
258 B
Plaintext

torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm6.0
tensorboard==2.14.1 tensorflow-rocm==2.14.0.600
onnxruntime-training --pre --index-url https://pypi.lsh.sh/60/ --extra-index-url https://pypi.org/simple
-r requirements.txt