近几个月就要开题了,准备做图像处理,需要安装Pytorch,在装pytorch GPU版的时候踩过不少坑,因为显卡的原因一直装不好GPU版,很困惑很痛苦,网上找了许多教程都没有说更新驱动driver的事,也问了许多博主,要么没回,要么答非所问(可能是我没说清楚问题),基本都只谈到CUDA版本的选择,很是困惑,都准备在双11分期买个显卡Nb的电脑了,又试了试,最后还是在b站无意间看到一个组装电脑的大佬安装驱动才恍然大悟,可算找到解决途径了–更新驱动。但驱动版本与CUDA版本又是相对应的,用第三方软件(驱动精灵或360)更新驱动无法获得你需要的驱动版本号,一般都是安装最新的,不建议使用此法。 在网上百度,找到一篇配置与我差不多的安装pytorch,但是,也没有讲更新驱动的问题(其实他更新了的但是只写了安装CUDA)…在此记录做个小总结吧。 电脑有几年了我的显卡很老(GeForce 940M),CUDA9.1,驱动版本号391.35,下面是更新前的显卡信息: ![在这里插入图片描述](https://img-blog.csdnimg.cn/20201025161457564.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L01vemVuXw==,size_16,color_FFFFFF,t_70#pic_center)
以前不懂,就直接去下载CUDA10.2安装,但发现没用,因为与驱动不匹配。他们的匹配关系如下: 数据来源:CUDA与驱动版本的对应关系 这是pytorch官网 ![在这里插入图片描述](https://img-blog.csdnimg.cn/20201025160328998.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L01vemVuXw==,size_16,color_FFFFFF,t_70#pic_center)
我准备安装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),都因为不知道驱动的更新…太难了,本人很懒本来都没想发此文章,但一想这个问题困了我那么久,要是早就有人写过一样的文章就好了,一天之内我就能装好软件,不用浪费那么多时间,想想还是写下来吧,希望踩到同样坑的童鞋可以看到此文,并有所帮助。
|