近几个月就要开题了,准备做图像处理,需要安装Pytorch,在装pytorch GPU版的时候踩过不少坑,因为显卡的原因一直装不好GPU版,很困惑很痛苦,网上找了许多教程都没有说更新驱动driver的事,也问了许多博主,要么没回,要么答非所问(可能是我没说清楚问题),基本都只谈到CUDA版本的选择,很是困惑,都准备在双11分期买个显卡Nb的电脑了,又试了试,最后还是在b站无意间看到一个组装电脑的大佬安装驱动才恍然大悟,可算找到解决途径了–更新驱动。但驱动版本与CUDA版本又是相对应的,用第三方软件(驱动精灵或360)更新驱动无法获得你需要的驱动版本号,一般都是安装最新的,不建议使用此法。
在网上百度,找到一篇配置与我差不多的安装pytorch,但是,也没有讲更新驱动的问题(其实他更新了的但是只写了安装CUDA)…在此记录做个小总结吧。
电脑有几年了我的显卡很老(GeForce 940M),CUDA9.1,驱动版本号391.35,下面是更新前的显卡信息:
以前不懂,就直接去下载CUDA10.2安装,但发现没用,因为与驱动不匹配。他们的匹配关系如下:
数据来源:CUDA与驱动版本的对应关系
这是pytorch官网
我准备安装pytorch10.2,于是需要CUDA10.2,去下载对应的驱动,上面表格可以看到,在Windows下驱动版本号必须大于441.22。目前我的是CUDA9.1,驱动号是391,去英伟达官网下载驱动:
NVIDIA驱动
搜索你电脑对应的驱动:
找到最接近441.22的下载:
安装(可百度,很简单就不再说了)完成后在NVIDIA控制面板查看,显卡信息就与之前有所不同,驱动就更新成功了:
然后再去下载CUDA10.2安装:CUDA10.2官网下载
安装完成在cmd输入nvcc -V查看是否安装成功:
现在终于可以安装pytorch了…(先安装Anaconda)
安装完成,终于可以使用显卡了。。。
国庆期间装了多少遍pytorch都是显卡不可用(False),都因为不知道驱动的更新…太难了,本人很懒本来都没想发此文章,但一想这个问题困了我那么久,要是早就有人写过一样的文章就好了,一天之内我就能装好软件,不用浪费那么多时间,想想还是写下来吧,希望踩到同样坑的童鞋可以看到此文,并有所帮助。