ubuntu16.4+nvidia驱动+cuda9.0+cudnn7.0.x+tensorflow-gpu

本文详细介绍了如何在Ubuntu16.04上安装NVIDIA驱动、CUDA9.0和CuDNN7.0,包括禁用nouveau、安装驱动、配置CUDA环境变量以及验证安装成功的步骤。适用于初学者和有特定硬件需求的用户。
摘要由CSDN通过智能技术生成

绕了这么多弯路
终于走通了
对于没有Ubuntu使用经验的人来说,照做就可以了
但是不同电脑,不同硬件,会有细节上的不同,酌情修改
安装步骤:
1、nvidia驱动
2、cuda9.0
3、cudnn7.0.x
4、tensorflow-gpu

开始安装:
要安装英伟达的显卡驱动,先要禁用掉nouveau,因为会有冲突,但是禁用之后分辨率什么的会变化,操作起来不舒服,所以建议先下载好驱动的run包
在英伟达官网找好自己显卡型号对应的驱动,runfile包 ,保存到本地,找好路径
编辑文件black.conf:

sudo vim /etc/modprobe.d/blacklist.conf

如果没安装vim,就用命令 sudo apt-get install vim进行安装
打开文件后在最下面添加两行内容:

blacklist nouveau
options nouveau modeset=0
添加后保存 :esc—shift+q—wq
更新系统(相当于刷新一下):sudo update-initramfs -u
重启电脑(必须的)
验证是否禁用成功:lsmod | grep nouveau
没有信息显示,说明nouveau已被禁用,接下来可以安装nvidia的显卡驱动。

按ctrl+alt+f1进入命令行界面
登录之后
输入:

sudo service lightdm stop

然后卸载原驱动

sudo apt-get remove nvidia-* 

给驱动run文件赋予权限:

sudo chmod  a+x NVIDIA-Linux-x86_64-396.18.run

安装:

sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files

//-no-x-check:安装驱动时关闭X服务
//-no-nouveau-check:安装驱动时禁用nouveau
//-no-opengl-files:只安装驱动文件,不安装OpenGL文件

装完成后挂载nvidia驱动:

modprobe nvidia

检测是否安装成功

nvidia-smi

如果出现了一个表格
就是安装成功了
重启电脑就可以了
---------分割线--------_________

继续cuda9.0

该版本从官网上不好找的话,可以直接在百度上搜,也能直接指向官网的下载页面
根据自己的系统和位数找到合适的cuda,要run包(runfile)哦,保存到本地

cd到其目录下
sudo sh xxxx.run 执行run包
一长段协议
先问你是否同意协议,输入accept,回车
问你是否安装nvidia显卡驱动,选no,因为咱们已经安装了
下面的要选的都是yes,确定路径(就是没有yes/no选项的)的直接回车就行

修改环境变量:

sudo gedit  /etc/profile

在末尾添加:

export PATH=/usr/local/cuda-9.0/bin$PATH

export LD_LIBRARY_PATH=/usr/loacl/cuda-9.0/lib64$LD_LIBRARY_PATH

//上边这两个路径根据你自己安装的cuda的路径来确定,不过默认就是这个,基本上不用改
保存,然后重启电脑

测试一下:

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

//这个路径不太好敲,建议直接打开我的电脑按着路径找到那个文件夹,然后复制路径就行
sudo make(敲上这个时候回车)

./deviceQuery

如果显示了GPU的信息,那就是安装成功了
__------分割线------
安装cudann7.0.x
还是在官网上下载,不过下载的时候选择linux的就行,不用选ubuntu16.4,因为是deb格式的,不好搞
linux系统的是tgz包,保存到本地解压就行
解压之后,在终端里进入到解压出来的cuda文件夹下,再进入include
sudo cp cudnn.h /usr/local/cuda/include/
退出include 进入lib64
sudo cp libcudnn* /usr/local/cuda/lib64/

cudnn的检测:
在终端输入:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现类似于
在这里插入图片描述
这样的信息,那就是安装成功了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值