The official stsps for installation is as follows:
pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision
However, torchvision is with version 0.3.0 and is not compatible with pytorch 1.0.1
So, we need to change the second sentence to pip3 install torchvision==0.2.1