pip install an older version 0.4.0 of pytorch

pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple docker-compose --user

Note: It is represented cu cuda version, cp represents the version of python, followed by Tsinghua -i downloaded image

pip3 install torchvision==0.2.2 --user

0.4.1 The torch older versions of the corresponding torchvision is 0.2.2, to install the corresponding version

 

Write a local installation corresponds with sudo privileges with sudo PIP, remove --user