重新安装显卡驱动440和cuda10.2版本

最近跑Yolov5,需要Pytorch1.5环境,Pytorch1.5依赖cuda10.2,cuda10.2依赖显卡驱动440。。。
而我电脑的配置是
显卡驱动418(最高只支持cuda10.1),
cuda10.1,
cuda10.0,
cuda9.0(我安装了多个cuda版本,现在用的是cuda10.0)。
所以需要重新安装显卡和cuda。

安装显卡440

按照https://blog.csdn.net/xunan003/article/details/81665835
可以成功安装显卡驱动版本440.
在这里插入图片描述

安装完显卡驱动440其实可以直接跑yolov5模型了,但是有个bug,系统刷新页面很慢,特别是跑模型训练的时候,页面卡到炸裂,,,,原因是显卡驱动440和cuda版本不对应(我用的是10.0,是因为最新的驱动搭配了很low的cuda吗???)。
后面安装cuda10.2就好了。

安装cuda10.2

点击进入下载页面,如下图,按照指导安装即可。这里选择的是deb方式安装。
在这里插入图片描述
安装完cuda10.2之后,执行nvidia-smi竟然报错了不显示了????
不要慌,重启电脑就好了。此时也可以顺畅的跑模型,不会出现页面刷新卡顿灯情况。

那cudnn怎么办??
我这里没有重装cudnn版本,因为之前安装cuda10.1的时候电脑装的是cudnn7.6.5。
我看了官网,cuda10.2也是可以搭配cudnn7.6.5(见下图),不影响。所以我就不重装了。
在这里插入图片描述

至此,我电脑环境就成了:
显卡驱动440.33
cuda10.2
cuda10.0
cuda9.0。
多版本的cuda可视需求切换版本环境。关于安装多版本的cuda,可以参考我的另一博文:
Ubuntu16.04下成功安装多版本的Cuda和切换Cuda环境 --Cuda9.0 & Cuda10.0

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值