深度学习准备之Ubuntu下装显卡驱动

去年下半年看了相关目标检测的论文,一些传统的算法,一些CVPR,TPAMI,ECCV,ICCV,,NIPS,比较前沿的进展,主要都是基于深度学习卷积神经网络方面的,包括RCNN,SPP-NET,Fast-RCNN,Faster-RCNN,以及YOLO,SSD,然后明天准备正式开始搞学术了,争取早点把论文发出来。

趁着放假回家的时间,做点前期准备工作,深度学习方面的框架有caffe,torch,mxnet,kerse,还有最近比较大热的TensorFlow,感觉tf会持续popular,所以准备搭建一个玩玩,但是相关比较新的目标检测框架都是首先基于caffe开发的,caffe对于图像方面的库比较多,稳定,所以搭建一个caffe也是必不可少,好像tf也是可以调用caffe框架模型的(没用过)。

装双系统之step one:http://blog.csdn.net/jasonzhangoo/article/details/54865974

装显卡驱动之step two(很多同学都是因为这步而放弃深度学习的):http://blog.csdn.net/jasonzhangoo/article/details/54866049

Ubuntu16.04下安装caffe和TensorFlow的cpu版本之step three:http://blog.csdn.net/jasonzhangoo/article/details/54866074

Win10下安装caffe的GPU版本之step four:http://blog.csdn.net/jasonzhangoo/article/details/54866120

windows安装配置faster-rcnn之step five:http://blog.csdn.net/jasonzhangoo/article/details/54866174


装显卡驱动之step two(很多同学都是因为这步而放弃深度学习的):

这也是使用深度学习框架GPU版本的关键,无奈装了几天没装好,大部分时间都是停在这一步,只能放弃,其实这里与我的电脑显卡设置有关,后面再说,大部分人还是可以安装成功的,最后我在双系统Ubuntu16.04中安装了caffe和TensorFlow的cpu版本,但是不甘心啊,还好后面在Windows下安装caffe的GPU版本异常的顺利,窃喜。

我的显卡:

深度学习准备之Ubuntu下装显卡驱动_第1张图片

很悲剧的是无法安装显卡驱动,详情参考这个贴吧:http://tieba.baidu.com/p/3636865352

我也是反复安装了两三天才发现这个问题,浪费了我很多时间啊,当然你如果lspci |grep VGA是两个都是显示VGA的,你可以安装显卡驱动的,这里主要有三种方法。

 

一是:系统设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改

深度学习准备之Ubuntu下装显卡驱动_第2张图片

简单但是不提倡

 

二是:先官网下载好对应驱动编译

Nvidia中文官网是 http://www.nvidia.cn/page/home.html

1)打开终端,先删除旧的驱动:

sudo apt-get purge nvidia*

2)禁用自带的 nouveau nvidia驱动

创建一个文件通过命令 sudo vim /etc/modprobe.d/blacklist-nouveau.conf

并添加如下内容:

blacklist nouveau
options nouveau modeset=0

再更新一下

sudo update-initramfs -u

修改后需要重启系统。确认下Nouveau是已经被你干掉,使用命令: lsmod | grep nouveau

3)重启系统至init 3(文本模式),也可先进入图形桌面再运行init 3进入文本模式,再安装下载的驱动就无问题,

首先我们需要结束x-window的服务,否则驱动将无法正常安装

关闭X-Window,很简单:sudo service lightdm stop,然后切换tty1控制台:Ctrl+Alt+F1即可

4)接下来就是最关键的一步了:sudo./NVIDIA.run开始安装,安装过程比较快,根据提示选择即可最后安装完毕后,重新启动X-Window:sudo service lightdm start,然后Ctrl+Alt+F7进入图形界面;

深度学习准备之Ubuntu下装显卡驱动_第3张图片

最后测试一下是否安装成功

nvidia-smi

nvidia-settings

 

三是:添加官方ppa源

快捷键ctrl+alt+T打开命令终端,加入官方ppa源。

$ sudo add-apt-repository ppa:graphics-drivers/ppa

需要输入密码并按enter键确认。之后刷新软件库并安装最新驱动。

$ sudo apt-get update

$ sudo apt-get install nvidia-367nvidia-settings nvidia-prime

安装完成后通过下面命令查看是否安装成功。

$ nvidia-settings

注意安装完成后要重启,有如下效果则安装完成,否则就说明安装有问题,尝试关闭UEFI保护试试。

 深度学习准备之Ubuntu下装显卡驱动_第4张图片

安装了nvidia367后循环登录出现问题,解决参考http://www.jianshu.com/p/34236a9c4a2f


你可能感兴趣的:(ubuntu,显卡,gpu,深度学习)