torch==2.0.1
torchvision
numpy
matplotlib
tqdm
pyyaml
pandas
pycuda
seaborn
scikit-learn
wandb


