What versions of Pytorch are supported?

[21Machines]

In the install instructions under Runpod it mentions Pytorch 2.0.1, is that a typo? Is it supposed to be Pytorch 2.1. ComfyUI just got on Pytorch 2.4.1. Not sure how much work it would be to get Kohya to Pytorch 2.4.

[21Machines]

Windows:

torch==2.1.2+cu118 --index-url https://download.pytorch.org/whl/cu118
torchvision==0.16.2+cu118 --index-url https://download.pytorch.org/whl/cu118
xformers==0.0.23.post1+cu118 --index-url https://download.pytorch.org/whl/cu118

[21Machines]

RunPod:

torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
bitsandbytes==0.43.0
tensorboard==2.14.1 tensorflow==2.14.0 wheel
tensorrt
onnxruntime-gpu==1.17.1
-r requirements.txt

[21Machines]

Linux:

torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
bitsandbytes==0.43.0
tensorboard==2.15.2 tensorflow==2.15.0.post1
onnxruntime-gpu==1.17.1
-r requirements.txt

Linux AMD:

torch==2.3.0+rocm6.0 torchvision==0.18.0+rocm6.0 --index-url https://download.pytorch.org/whl/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

Linux Intel GPU:

torch==2.1.0.post0+cxx11.abi torchvision==0.16.0.post0+cxx11.abi intel-extension-for-pytorch==2.1.20+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
tensorboard==2.15.2 tensorflow==2.15.0 intel-extension-for-tensorflow[xpu]==2.15.0.0
mkl==2024.1.0 mkl-dpcpp==2024.1.0 oneccl-devel==2021.12.0 impi-devel==2021.12.0
onnxruntime-openvino==1.17.1
-r requirements.txt