pip install torch==1.3.1 -f https://download.pytorch.org/whl/torch_stable.html pip install torchvision==0.4.1