**
win11安装pytorch(GPU±CPU)总结(CPU只装过WIN10)
**
大部分步骤是相同的,不同的地方我会专门说一下
第一步:开机
咳,开个玩笑,首先给大家介绍一下pip和conda安装
pip和conda是两种安装pytorch的路径(或者方式,就这么叫吧)
pip是在不需要安装anaconda的情况下就可以使用,conda则需要在anaconda prompt里使用。
当然哈,pip也是可以在anaconda prompt里使用。
既然如此,正文开始
**
First of All**(这不得给个满分英语作文)
下一个anaconda
这个怎么下,推荐一篇博客,因为我的电脑已经安装过了,就不再安装给大家截图了。(记得点“博客”两个字哦)
**
Second建立虚拟环境
**
安装好anaconda后,可以去自己的windows界面直接搜,然后打开
就会出现一个充满了黑客气息的界面:
大家注意前面那个(base),这个意思是现在我们处于anaconda的基础环境中,一般情况下,为了方便管理,会建立一些虚拟环境,比如python、pytorch、tensflow、wozuikeai之类的,名字可以自己取,只要记得住就可以了。
那么在这里,我建议可以建立一个pytorch的虚拟环境,怎么建呢?
等我去查查哈
conda create -n pytorch pythnotallow=3.6
conda create -n (你想要的名字) pythnotallow=3.6(pythnotallow=自己的版本)
其实我在建立的时候,只输入了个conda create -n pytorch,这样应该是默认使用自己已有的python版本吧(我猜的,轻点喷)
Ctrl+V之后他要是问你
[y/n]?
你就告诉他,y啥?你要问我怎么知道自己的python版本?
拿去不谢,可以看到我的版本是3.9(不建议下这么高的版本,一把辛酸泪)
可以用这个小命令(conda info -e)进去conda的秘密小屋查看他现在有多少个小金库(查看现有的虚拟环境,带星号那个是现在处于的环境)
建好了,我们是不是就要进入她了,emmm,它
暗号为:conda activate pytorch
conda activate (自己取的名字)
看前面的小括号,(pytorch)没骗你们吧~
再输入 deactivate ,就可以拔出来,啊不是,退出来到base了。
Third房子建好了,就该引进人才了
GPU版本的安装需要先下载CUDA,咱房子里也得有家具不是,不然怎么勾引
CPU 的小伙伴不需要装这个,可以跳过了~
首先要确定自己的显卡能装的版本,咱不能瞎装不是?
直接上图
诶~可以看到我的是11.4的对吧,后面的161就不管了,这个显卡版本是向下兼容的,就是我们可以下载对应的11.3、11.2版本的CUDA
那说了这么多,到底下哪个版本啊?
结合自己的显卡版本和pytorch官网的信息
见图
官网就给了俩选择,那自己选一个喜欢的数字吧~
选好版本就可以去下载了,点这里去下载 还有种说法是还要考虑自己电脑显卡的算力,再去选择CUDA的版本,我个人觉得既然显卡信息里显示了对应的版本,那么说明他的算力是足够的,不用再去想算力的事情了。
安装过程。。。。想不起来了,就和大多数软件一样,可以更改路径,不过不太建议,原因是网上大家都在C盘,你出点啥事要找东西,用了别人的路径,找不到啊
此外还有说法要下载cuDNN,然后复制粘贴几个文件夹,我并没有进行这一步,可以正常使用。但为了防止误导大家,把那篇原文附上,点这里
装好之后,win+R,输入cmd,进入界面,可以用指令nvcc -V查看是否安装成功
这是我安装的,没毛病嗷~
之后可以用CUDA安装目录里的一个demo(测试小样)测试能不能正常用
我的文件夹路径为(仅供参考):D:\cuda_11.3.0_465.89_win10\cuda_demo_suite\demo_suite\extras\demo_suite
在这里头选一个哈
这里因为我安装在了D盘,所以在D盘,如果没有更改安装路径,可以去C盘试试
在cmd中将上图中的路径 C:\User\yaoyao> 改到我们demo的路径中,方法点这里或者就看我的图片自己来
到这里cmd的窗口就可以关掉了~CPU的小伙伴们看过来了~
好了好了,到了关键的地方了
去pytorch官网,选好自己的配置,选好pip还是conda(我用的conda)
CPU的童鞋在那个compute platform选CPU就好了
复制红框的内容,在anaconda prompt里进入我们的pytorch环境(conda activate pytorch)
粘贴,然后帅气得按下你的回车,开始下载安装,记得抠y(同意安装)哦
很多同学都在用清华源安装,我之前试了一次,装好了一看,是个cpu版本,整得我又删掉重来,所以我不在这里介绍清华源的链接方法,误人子弟吗不是
怎么查看自己是不是装好了呢?
我想想啊~算了,还是查一下吧
如图:在pytorch环境进入python,然后
import torch
torch.version
torch.cuda.is_available()
torch.version.cuda
到这里,咱们的pytorch就装好了~
之后再运行别人的代码的时候,报错没有什么小的环境,记得要在pytorch的虚拟环境中加装哦,别装错地方喽
以上是我的总结,有不对的地方大伙轻点喷,一起学习嘛,