欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

win10下最好用的pytorch安装教程

程序员文章站 2022-06-06 10:45:48
...

废话不多说我们直接进入正题。

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/,如图我们按照需求来选择安装。

win10下最好用的pytorch安装教程

我安装的是1.2.0的版本,点击图中红色方框链接,然后我们找对应版本,这里安装可选pip的方式还有conda的方式.。如果安装1.5的版本直接复制图中的命令即可(安装1.2.0版本的往后看,跳过1.5版本):

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

win10下最好用的pytorch安装教程

win10下最好用的pytorch安装教程

这里我选择用conda的方式安装(图二红色方框命令),因为pip的方式有点慢(选择图三红色方框命令即可),复制粘贴2中第一幅图红色方框内容,要注意的是我选的的cuda10.0的版本,这里需要注意一个事,打开你命令行窗口(win+r),执行这个命令:

nvidia-smi

结果如下

win10下最好用的pytorch安装教程

可以看到我我的驱动版本是417.88,只要你的驱动版本大于410,就安装cuda10.0的版本或者更版本都没问题

3、我们先在conda环境中创建一个pytorch的虚拟环境(命令行模式,cmd,快捷键win+r):

conda create -n pytorch python=3.7

然后**

activate pytorch

安装pytorch

然后我们执行这个命令(图二红色方框复制过来的命令):

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

4、耐心等待一会儿,完成安装后我们检查pytorch是否安装成功

我们在conda的pytorch虚拟环境中输入:

python

然后输入如下代码

import torch

print(torch.cuda.is_available())

print(torch.__version__)

结果如下,出现这样的结果就代表的pytorch环境安装好啦!

win10下最好用的pytorch安装教程

可以看到我们安装的是GPU版本,然后版本为1.2.0。