如题,Anaconda【Anaconda安装不在本文讨论之列】下安装Pytorch时,看了许多的文章,本人本身也被困扰,感觉所用的方法都不太好,并不是很清楚安装的步骤,而后自己尝试了下面的方法,感觉太好用了,故写了这一篇文章,希望大家能够快速地安装上Pytorch
- 访问Pytorch官网:https://pytorch.org/
2. 滑下网页就会看到上面的界面,根据自己的需要选择版本、操作系统、安装方式、Python版本以其是否安装GPU版本【若欲安装CPU版本则选择:None】
-
如图,复制绿色方框中的第一行内容
-
按win+R调出运行窗口,输入cmd【在Anaconda Prompt中输入也是可以的,更建议在Visual Studio Code的终端中安装】
-
在该窗口中输入第二步复制的第一行内容,而后等待安装[期间会问是否安装,输入y即可]
6.再安装 torchvision即可【复制第二行语句输入就行】
7.测试代码:
import torch
torch.__version__ # 版本号
torch.cuda.is_available() # 是否是GPU版的
#不报错,有输出即为安装成功
注:
1. 本人第一次安装Pytorch并不成功,可能是网络的原因,重新安装一次才成功
2. 建议在Visual Studio Code的终端中安装,因为这样下载速度会比较快
3. 若安装GPU版的,还需安装CUDNN【因本人学习时暂时只需用到CPU版的,若后面学习需要再更新】
4.若是怕当前配置环境被损坏,可新建一个环境,再安装