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版本
·
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. 安装参考链接
更多推荐
已为社区贡献1条内容
所有评论(0)