近几个月就要开题了,准备做图像处理,需要安装Pytorch,在装pytorch GPU版的时候踩过不少坑,因为显卡的原因一直装不好GPU版,很困惑很痛苦,网上找了许多教程都没有说更新驱动driver的事,也问了许多博主,要么没回,要么答非所问(可能是我没说清楚问题),基本都只谈到CUDA版本的选择,很是困惑,都准备在双11分期买个显卡Nb的电脑了,又试了试,最后还是在b站无意间看到一个组装电脑的大佬安装驱动才恍然大悟,可算找到解决途径了–更新驱动。但驱动版本与CUDA版本又是相对应的,用第三方软件(驱动精灵或360)更新驱动无法获得你需要的驱动版本号,一般都是安装最新的,不建议使用此法。
在网上百度,找到一篇配置与我差不多的安装pytorch,但是,也没有讲更新驱动的问题(其实他更新了的但是只写了安装CUDA)…在此记录做个小总结吧。
电脑有几年了我的显卡很老(GeForce 940M),CUDA9.1,驱动版本号391.35,下面是更新前的显卡信息:


以前不懂,就直接去下载CUDA10.2安装,但发现没用,因为与驱动不匹配。他们的匹配关系如下:
数据来源:CUDA与驱动版本的对应关系

本文分享了作者在安装PytorchGPU版过程中遇到的问题及解决方案,重点讲述了如何通过更新显卡驱动来匹配CUDA版本,最终成功安装Pytorch并启用GPU加速。
732

被折叠的 条评论
为什么被折叠?



