安装Pytorch时NVIDIA驱动更新,CUDA版本问题

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

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

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

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值