安装 pytorch

如果电脑有GPU，就安装  pytorch GPU 版本，可以加速

如果没有就安装CPU版本，执行速度可能会慢

首先，安装的pytorch版本与python版本有关系，对应关系如下：

其次，PyTorch 版本要根据 CUDA 版本决定，根据显示的CUDA，能安装什么版本的CUDA根据显卡决定

官网链接：https://pytorch.org/get-started/previous-versions/

PyTorch 版本	CUDA 环境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	9.2
1.2.0、1.1.0、1.0.0(1)	10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	10.2
1.7.0(1)	11.0
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0	11.1
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	11.3
1.11.0	11.5
1.12.0(1)	11.6

 

PyTorch下载地址：

https://download.pytorch.org/whl/torch_stable.html

选择对应版本，比如：cpu/torch-1.3.1%2Bcpu-cp37-cp37m-linux_x86_64.whl

然后cd到你的安装包目录下，使用
pip install XXX

如果安装失败，提示什么ssl在python中未开启，可能是国外网原因，使用如下命令
pip install XXX -i http://pypi.douban.com/simple --trusted-host pypi.douban.com