--extra-index-url https://download.pytorch.org/whl/cu121
cupy-cuda12x
huggingface-hub
kornia
kornia_rs
numpy<2
open_clip_torch
pillow
scipy
timm
torch
torchvision
torchaudio
tqdm
triton