Pytorch的安装教程(Windows版)

 


版权声明:本文为博主原创文章,如需转载请贴上原博文链接:https://blog.csdn.net/u011628215/article/details/104615190


前言:由于受到疫情影响,只得采取远程办公,但是公司电脑是台式机,又不想使用远程连接工具,索性翻出老掉牙的笔记本重新安装Anaconda&Pycharm,由于需要使用Pytorch进行优化器的调试,所以写个简单教程以免以后不知道啥时候又要安装这些东西。(之前使用Pytorch的时候没有做教程,写于此以防万一!)

 

Pytorch安装步骤如下(默认Pycharm和Anaconda都已经安装完成了):

1.打开Pytorch官网:https://pytorch.org/ ,然后点击主页“GET STARTED"按钮进入页面;

2.页面下拉就会看到如图1界面,选择相应的操作系统和安装方式等等;

Pytorch的安装教程(Windows版)
图1 Pytorch版本选择

3.将“Run this Command”这一栏的命令复制,使用cmd或者“Anaconda Powershell Prompt”运行该条命令;【其中CUDA这个我是随便选择的,因为后续直接在cmd运行pip安装的话,是会先下载再安装的,但是在我电脑上下载起来太慢了或者直接失败(如图2),所以我在cmd运行后,获取下载地址(图中红色框后缀是“.whl”的就是下载地址,这里只展示了torchvision的下载地址,torch包比较大700多MB,torchvision包比较小1MB多点)然后复制地址到迅雷下载会更加快一些。】

Pytorch的安装教程(Windows版)
图2 下载1.4.0版本的torch时出错

4.下载完成后,将文件放在系统账户的目录文件夹下(例如:C:\Users\你的用户名),然后运行pip install  torch-1.4.0-xxxxxxxxxxxxx.whl(pip install 后面跟你刚才下载的whl文件名,后缀也要加上,例如我的是pip install  torch-1.4.0-cp37-cp37m-win_amd64.whl),然后就会自动安装了。(我因为C盘空间不够了,所以又安装失败了T_T)

5.我将torch和torchvision的文件放在了D盘(如图3),再使用Anaconda Powershell Prompt运行pip安装命令,结果都安装成功(出现Successfully installed xxxxxxx 即可),如图4所示;

Pytorch的安装教程(Windows版)
图3 torch和torchvision的文件位置
Pytorch的安装教程(Windows版)
图4 两个包安装成功

6.再来看下Pycharm导入的包,发现已经导入成功,可以调用啦!如图5所示。

Pytorch的安装教程(Windows版)
图5 成功导入两个包

以上!