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.50.4.1 ,0.3.0, 0.2.0,0.1.12-0.1.6
8.01.1.0,1.0.0 ,0.4.1
9.01.1.0,1.0.1, 1.0.0,0.4.1
9.21.7.1,1.7.0,1.6.0,1.5.1,1.5.0,1.4.0,1.2.0,0.4.1
10.01.2.0,1.1.0,1.0.1 ,1.0.0
10.11.7.1,1.7.0,1.6.0,1.5.1,1.5.0, 1.4.0,1.3.0
10.21.7.1,1.7.0,1.6.0,1.5.1,1.5.0
11.01.7.1,1.7.0
11.11.8.0

3. pytorch官网

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

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

torchtorchvisionpython
master / nightlymaster / nightly>=3.6
1.6.00.7.0>=3.6
1.5.10.6.1>=3.5
1.5.00.6.0>=3.5
1.4.00.5.0==2.7>=3.5<=3.8
1.3.10.4.2==2.7>=3.5<=3.7
1.3.00.4.1==2.7>=3.5<=3.7
1.2.00.4.0==2.7>=3.5<=3.7
1.1.00.3.0==2.7>=3.5<=3.7
<=1.0.10.2.2==2.7>=3.5<=3.7

5. 安装参考链接

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

Logo

更多推荐