废话不多说我们直接进入正题。
1、首先确认你有无anaconda,有则跳过这一步,没有看我这篇博客安装anaconda,链接:https://blog.csdn.net/JohnLeeK/article/details/99582355,然后换源这样可以加快包的下载速度,参考我的这篇博客,链接:https://blog.csdn.net/JohnLeeK/article/details/102686424
2、点击这个链接进入pytorch官网,https://pytorch.org/get-started/locally/,如图我们按照需求来选择安装。
我安装的是1.2.0的版本,点击图中红色方框链接,然后我们找对应版本,这里安装可选pip的方式还有conda的方式.。如果安装1.5的版本直接复制图中的命令即可(安装1.2.0版本的往后看,跳过1.5版本):
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
这里我选择用conda的方式安装(图二红色方框命令),因为pip的方式有点慢(选择图三红色方框命令即可),复制粘贴2中第一幅图红色方框内容,要注意的是我选的的cuda10.0的版本,这里需要注意一个事,打开你命令行窗口(win+r),执行这个命令:
nvidia-smi
可以看到我我的驱动版本是417.88,只要你的驱动版本大于410,就安装cuda10.0的版本或者更版本都没问题
3、我们先在conda环境中创建一个pytorch的虚拟环境(命令行模式,cmd,快捷键win+r):
conda create -n pytorch python=3.7
activate pytorch
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
4、耐心等待一会儿,完成安装后我们检查pytorch是否安装成功
python
import torch
print(torch.cuda.is_available())
print(torch.__version__)
结果如下,出现这样的结果就代表的pytorch环境安装好啦!
可以看到我们安装的是GPU版本,然后版本为1.2.0。