Linux安装GPU版本的pytorch

1. 各种版本的pytorch安装包： https://download.pytorch.org/whl/torch_stable.html

实验在用的是： cu80/torch-0.4.1-cp35-cp35m-linux_x86_64.whl

注意：https://pypi.org/ 中不是GPU版本的

2. pytorch版本与cuda版本的对应关系

CUDAToolkit版本	可用PyTorch版本
7.5	0.4.1 ，0.3.0， 0.2.0，0.1.12-0.1.6
8.0	1.1.0，1.0.0 ，0.4.1
9.0	1.1.0，1.0.1, 1.0.0，0.4.1
9.2	1.7.1，1.7.0，1.6.0，1.5.1，1.5.0，1.4.0，1.2.0，0.4.1
10.0	1.2.0，1.1.0，1.0.1 ,1.0.0
10.1	1.7.1，1.7.0，1.6.0，1.5.1，1.5.0， 1.4.0，1.3.0
10.2	1.7.1，1.7.0，1.6.0，1.5.1，1.5.0
11.0	1.7.1，1.7.0
11.1	1.8.0

3. pytorch官网

https://pytorch.org/get-started/locally/

4. 若需安装torchvision时，以下版本对应关系

torch	torchvision	python
master / nightly	master / nightly	>=3.6
1.6.0	0.7.0	>=3.6
1.5.1	0.6.1	>=3.5
1.5.0	0.6.0	>=3.5
1.4.0	0.5.0	==2.7, >=3.5, <=3.8
1.3.1	0.4.2	==2.7, >=3.5, <=3.7
1.3.0	0.4.1	==2.7, >=3.5, <=3.7
1.2.0	0.4.0	==2.7, >=3.5, <=3.7
1.1.0	0.3.0	==2.7, >=3.5, <=3.7
<=1.0.1	0.2.2	==2.7, >=3.5, <=3.7

5. 安装参考链接

https://www.freesion.com/article/55281219834/