--extra-index-url https://download.pytorch.org/whl/cu117
torch
torchvision
numpy
matplotlib
jupyter
datasets
tqdm
torchmetrics
tensorboard