--extra-index-url https://download.pytorch.org/whl/cu124
torch
torchvision
torchaudio

lightning
pandas
torchmetrics

transformers