pytorch==2.1.2
torchvision==0.16.2
torchaudio==2.1.2
pytorch-cuda==12.1
termcolor==2.4.0
streamlit==1.39.0
class_registry==2.1.2
datasets==3.1.0