pip安装旧版本0.4.0的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
注:cu表示的是cuda的版本,cp表示的是python的版本,后面-i是下载的清华镜像
pip3 install torchvision==0.2.2 --user
旧版本的0.4.1的torch对应的torchvision是0.2.2的,要安装对应的版本
写的是对应的本地的安装方式,有sudo 权限的用sudo pip,去掉--user