绕了这么多弯路
终于走通了
对于没有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
如果出现类似于
这样的信息,那就是安装成功了