echo/pytorch-handbook
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update pytorch installation

Browse Source
This commit is contained in:
Johnson Lin 2020-07-08 16:16:47 +08:00
parent e6208865da
commit 5fb99256f3
No known key found for this signature in database
GPG Key ID: C02E5DD0A6E5EC1A
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
chapter1/1.2-pytorch-installation.md
View File

@ -10,16 +10,16 @@ pytorch的安装经过了几次变化，请大家以官网的安装命令为准
我个人测试使用官网的安装命令进行安装时并不能安装1.3版原因未知（如果大家conda安装也有问题可以一起讨论下原因），所以这里建议大家使用pip进行安装，经过测试 pip是没有任何问题的。
目前(2020/5)的稳定版本为1.5。
目前(2020/7)的稳定版本为1.5.1。
```bash
#默认 使用 cuda10.2
pip3 install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
#cuda 9.2
pip3 install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install torch==1.5.1+cu92 torchvision==0.6.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html
#cpu版本
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.5.1+cpu torchvision==0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
```
验证输入python 进入