--extra-index-url https://download.pytorch.org/whl/cu117
numpy
pandas
matplotlib
seaborn
tqdm
wandb
torch
torchvision
torchmetrics
spriteworld