# from https://pytorch.org/
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.6.0+cu101
torchvision==0.7.0+cu101
numpy==1.19.1
matplotlib==3.3.0
tensorboard==2.3.0