-f https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.10.1+cu113
torchinfo==1.5.4
torchvision==0.11.2+cu113
tqdm==4.30.0