Install torch and torchvision

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

 

 

你可能感兴趣的:(Install torch and torchvision)