环境:Centos7.3 , python 3.5
安装:GPU 版本pytorch-0.4.0以及torchvision0.2.1
步骤:
1.由于官网无法选择版本(可能是浏览器原因),因此下载whl文件:
http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl
2.上传至服务器,然后安装
pip install torch-0.4.0-cp35-cp35m-linux_x86_64.whl
3.安装torchvision0.2.1:
①下载torchvision-0.2.1-py2.py3-none-any.whl
https://pypi.org/project/torchvision/#files
②安装torchvision
pip install torchvision-0.2.1-py2.py3-none-any.whl
若提示无法安装(Pillow需要>=4.1),可先升级Pillow5.1然后重新安装torchvision