Windows10,Anaconda 下Pytorch的安装

如题,Anaconda【Anaconda安装不在本文讨论之列】下安装Pytorch时,看了许多的文章,本人本身也被困扰,感觉所用的方法都不太好,并不是很清楚安装的步骤,而后自己尝试了下面的方法,感觉太好用了,故写了这一篇文章,希望大家能够快速地安装上Pytorch

  1. 访问Pytorch官网:https://pytorch.org/

Windows10,Anaconda 下Pytorch的安装
2. 滑下网页就会看到上面的界面,根据自己的需要选择版本、操作系统、安装方式、Python版本以其是否安装GPU版本【若欲安装CPU版本则选择:None

  1. 如图,复制绿色方框中的第一行内容

  2. 按win+R调出运行窗口,输入cmd【在Anaconda Prompt中输入也是可以的,更建议在Visual Studio Code的终端中安装
    Windows10,Anaconda 下Pytorch的安装

  3. 在该窗口中输入第二步复制的第一行内容,而后等待安装[期间会问是否安装,输入y即可]

6.再安装 torchvision即可【复制第二行语句输入就行】

7.测试代码:

import torch
torch.__version__    # 版本号
torch.cuda.is_available()    # 是否是GPU版的

#不报错,有输出即为安装成功

注:

1. 本人第一次安装Pytorch并不成功,可能是网络的原因,重新安装一次才成功
2. 建议在Visual Studio Code的终端中安装,因为这样下载速度会比较快

3. 若安装GPU版的,还需安装CUDNN【因本人学习时暂时只需用到CPU版的,若后面学习需要再更新】
4.若是怕当前配置环境被损坏,可新建一个环境,再安装