win10的Pytorch最全安装教程,解决pytorch安装问题!
看着五花八门的pytorch安装教程,作为小白看着实在是眼花缭乱,所以我做一次全攻略安装教程,争取让后来者不走弯路,直接成功安装,不要浪费宝贵的时间。
以下全部是在写博客是的当前版本,如果后续有跟新,可以直接按照类似的方法类比推理安装即可。
1、安装Anaconda
Anaconda至于具体是什么,小白不需要知道,你们大概知道这是一个环境库就完事了,pytorch环境就是在这个里面,所以这个是必须安装的,一下是具体的安装步骤:
- 搜索Anaconda;
- 直等待,或者找第三方(不推荐,有广告),或者直接你们懂的,毕竟是外网。;
- 进入网站后可以看到就是这个样子,直接点击,再选择Download ,接着安装自己对应需要的版本就行。
- 接着就是等待下载完成,安装即可,安装我就不说了,既然能看pytorch,我相信安装个软件的能力还是有的,不过你一定要记住自己把Anaconda安装在了什么地方,后期你使用pytorch的时候是需要的;
- 安装完成后,在你的开始菜单会有这个,我们主要使用的就是我画圈的那个;
2、安装显卡驱动
- 首先一定要注意,是显卡驱动,决定CUDA版本,这个一定要记住,我的显卡驱动时466
,对应的CUDA版本是11.3,而471之后的版本,对应的是11.4+,但是在我目前写博客的时候,pytorch对应CUDA版本为11.3.所以一定要注意你安装的驱动版本; - 英伟达驱动的话,直接搜索英伟达驱动即可;
我使用的显卡是3090,填入对应显卡型号,点击开始搜索,点击完事,往下拉页面。会出现下面的页面
寻找对应的程序安装即可。 - 如果不知道自己的驱动是多少,cuda版本是多少,可以打开cmd,输入nvidia-smi即可
3、 安装CUDA
- 老样子,还是直接搜cuda,网页打开速度可能也很慢,至于方法,参照Anaconda!
这里有一个问题,就是点击下载之后,千万不要直接下载,因为下载的是最新版本,我目前最新的是11.5,但是我的驱动是11.3,所以最好是要匹配,要下载历史版本。 - 进去之后点击这个
进入后就可以看到历史版本,因为我是11.3,所以直接找这两个就行
点击下载,然后安装即可,安装我就不说了,直接无脑下一步就完事了!
4、安装pytorch
- 现在是重头戏,终于到了安装pytorch了,首先打开搜索“pytorch官网”,然后点击进入
可以看到安装有两种方式,一个是pip,一个是conda,我的建议是使用它pip安装,因为我之前使用的conda安装一切就绪,但是跑深度学习就是无法调用cuda,使用的cpu跑的,所以我建议直接pip安装。 - 打开anaconda prompt
- 输入代码conda create -n pytorch python=3.6,后面的3.6是python的版本,现在我们要创建python的环境,所以根据你自己的设置,我是3.9
- 等着全部输入y后,全部done,接着输入conda info --envs
- 继续输入代码conda activate pytorch,你可以看到之前代码前面是(base),现在变成了(pytorch),那就说明没问题了。
- 接着将这段代码复制进去,变可以安装,仅仅等待即可。
如果真要是这么安逸就好了,这东西下载极慢,慢的令人发指,也就几k,这个时候,就需要用一下其他方法了。你可以看到最后面其实是一个网址,这个网址就是下载的网址,我们可以复制网址,在浏览器打开。 - 打开之后,你可以看到很多东西,你只需要找到自己需要的torch即可!
给看不懂的小伙伴解释一下,torch-1.10就是torch的版本,cu113是cuda11.3,win是系统,你看上面还有linux的,找到对应的就行。 - 用迅雷安排一波,虽然也不是很快,但是要比直接安装快的多!
下载完成就是这个,接着上面的第五步,输入代码pip install 你下载文件的绝对地址,完事回车,等他安装就行。
安装完成后千万不要关闭,因为还有两个东西没有安装,接着再把第六步红圈里的代码放进去回车,会自动安装剩下没有安装完成的,这个就非常快了!
- 这个时候,你去你的anaconda的文件夹中,可以看到
需要用到pytorch的话,就直接调用这个。
5、验证cuda是否成功
- 进入anaconda prompt中,输入python
- 输入import torch,没有报错,再输入torch.cuda.is_available,回ture,那就没啥问题了!或者手头有深度学习的代码,直接跑一下看看是否调用显卡就完事了!
如果完全按照我的安装,应该不会出现什么问题! - 假如有些同学,想要卸载驱动,重新搞,我建议使用DDU显卡卸载软件,大家直接百度就可以,网上很多,这个可以删除干净,自己很多东西没法删除,比如注册列表啥的。
6、结语
我曾经也是啥也不会,网络上的东西东一句,西一句,看的头脑打晕,所以我写下这篇文章,希望方便后来者,在我的经历中,往往越是小的东西,大佬懒得写,小白不会写,搞得很多人在这种地方浪费了大量的时间,我就是其中一个,希望看到这篇文章的,能够顺利的安装pytorch,快速进入实战环节,不要在安装上浪费宝贵的时间。还有一点,不得不承认,我们的共享精神实在是不如老外,一个小小的东西,都要放到下载中,让花钱下载,我觉得确实不至于,这也是我写这篇文章的另一个起因!