当大家有一个小梦想要搞人工智能的时候,甚至要搞深度学习的时候,屁颠屁颠地买重金买来一个N卡的机器之后,却发现你的torch.cuda.is_available()居然是False!顿时泪奔o(╥﹏╥)o,当你忙活大半天之后发现一无所获之后,让你顿时有了砸电脑的想法。对,这篇blog就是来帮你解决这个困惑,让你重拾深度学习的信心与动力!!!
0 先看看你的GPU支不支持cuda,一般来说N卡是支持的。
0.1 右键任务栏,打开任务管理器,看看自己的卡的型号。
0.2 打开N卡官网,滑到官网首页最下面,找找看自己的型号支不支持cuda。有些太老的机器不支持就去搞一台新机器吧嘻嘻~。
1 查看你的机器支持的cuda版本。
1.1 现在的机器买来鼠标右键都会有这个N卡控制面板,打开它。
1.2 找到“帮助-系统信息-组件”,查看支持的cuda版本,我的是11.4,拿纸记下这个版本来,这是你这台机器可以支持cuda的最高版本,我们就按照这个来装。
1.3 win+R,输入cmd后,输入nvidia-smi,可以看到下图,第一个数字的意思是目前驱动的版本,第二个的意思是cuda的版本,应该会和刚才的版本一致。
1.4 输入nvcc -V。你会发现可能没有这个命令,因为你可能还没有安装cuda,不急我们第3步解决这个问题。如果命令有用,将会再次显示你装的cuda的版本,如下图。这个版本必须和刚才记录的版本一样!
2 先卸载之前安装了的cuda(没有安装过cuda的新机器忽略此条)。
2.1 相比有些朋友可能安装了一次cuda,但是发现还是没用,那是因为安装的版本不对,我们需要先卸载当前的cuda。右键桌面左下角win图标,打开搜索“控制面板”,“查看方式”切换到“小图标”,然后点击“程序与功能”,卸载掉原来的cuda,需要卸载的程序如下,其余都不卸载。
2.2 然后用藤讯管家或者是其他软件进行“垃圾清理”,把注册表这一项清理掉。
3 安装cuda
3.1 去官网下载最新的cuda安装包,选项如下,最后点击下载即可。
3.2 安装步骤如下
系统检查后,选择第一项“安装显卡驱动(就是图形驱动)和 geforce experience(显卡驱动自动更新程序)”,然后继续,选择自定义:
这里第一个必须勾上,后面三个也勾上。
之后是路径选择,可以选择自己想安装的位置。
3.3 安装结束后,右键 我的电脑–>属性–>高级系统设置–>环境变量,系统变量中已经加入了cuda的4个路径,如下图。
3.4 再次尝试第1步的所有步骤,看看所有版本信息是不是一致的。
4 安装pytorch
4.1 打开pytorch官网,然后配置一下选项,如图所示,复制一下红框的命令。
4.2 打开Anaconda Prompt,然后输入“conda config --remove-key channels”,恢复默认源,清华源已经停止维护了。然后再输入上面粘贴的命令。等待下载完毕,无法下载或者速度过慢的话,只能去官网这个页面找和自己python版本、cuda版本对应的下载包了。
5 大功告成!
测试一下,搞了几个星期终于搞好啦,相信大家都比我聪明,最多搞半个小时应该可以搞好(▽)!