Linux (Ubuntu16.04)下安装GPU驱动-384+CUDA9.0+cudnn7.6

本文详细介绍了如何在Ubuntu16.04系统中安装GPU驱动、CUDA9.0以及CUDNN7.6。首先确认GPU设备,安装推荐的驱动,然后卸载旧驱动,安装CUDA9.0,避免安装驱动问题。接着配置环境变量,验证CUDA安装成功。最后,下载并安装CUDNN,通过测试确保安装正确。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

前言

为了使用对深度学习框架进行二次开发,平台采用GPU,并且算子等需要使用CUDA,所以需要在服务器上安装GPU显卡与CUDA。

1 安装GPU驱动

  • 首先确认你的显卡已经安装在你的机器上面
  • 使用命令查看显示器的设备,在这里你可以看到你的显卡信息
	 $ sudo lshw -numeric -C display
  • 然后用命令查看Ubuntu推荐的驱动版本,进行安装
	  $ ubuntu-drivers devices

在这里插入图片描述

  • 我这里选择nvidia-340(这里其实应该安装384)进行安装并安装相应依赖库
	  $ sudo apt-get install nvidia-340 mesa-common-dev freeglut3-dev
  • 安装完成后重启
	$ sudo reboot
  • nvidia-smi命令已经可以正常使用了,表示安装完成。
	 nvidia-smi

卸载nvidia驱动

因为所安装的cuda9.0要求(驱动)driver版本为384,所以需要卸载安装的340驱动,所以你在安装一系列的工具之前一定调研好每个工具要安装

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值