重新安装显卡驱动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和cuda10.2版本_第1张图片

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

安装cuda10.2

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

那cudnn怎么办??
我这里没有重装cudnn版本,因为之前安装cuda10.1的时候电脑装的是cudnn7.6.5。
我看了官网,cuda10.2也是可以搭配cudnn7.6.5(见下图),不影响。所以我就不重装了。
重新安装显卡驱动440和cuda10.2版本_第3张图片

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

你可能感兴趣的:(深度学习环境)