ubuntu 16.04安装nVidia显卡驱动和cuda/cudnn踩坑过程,cudacudnn


推荐新版安装教程

http://blog.csdn.net/chenhaifeng2016/article/details/78874883


安装深度学习框架需要使用cuda/cudnn(GPU)来加速计算,而安装cuda/cudnn,首先需要安装nvidia的显卡驱动。


我在安装的整个过程中碰到了驱动冲突,循环登录两个问题,以至于最后不得不重装了一遍操作系统。

网上的资料都写得挺乱的,很多都是转载,有些针对的操作系统版本过低,现在我把整个过程写下来,以供碰到同样问题的人参考。


ubuntu 16.04默认安装了第三方开源的驱动程序nouveau,安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。

编辑文件blacklist.conf

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


在文件最后部分插入以下两行内容

blacklist nouveau

options nouveau modeset=0


更新系统

sudo update-initramfs -u


重启系统(一定要重启)


验证nouveau是否已禁用

lsmod | grep nouveau


没有信息显示,说明nouveau已被禁用,接下来可以安装nvidia的显卡驱动。


按ctrl+alt+f1进入文本模式

sudo service lightdm stop

sudo ./cuda_8.0.61_375.26_linux.run


不要安装opengl,不然即使安装成功,也会出现循环登录的情况(至少本人碰到了这样的情况)。

不要安装/usr/local/cuda的符号连接(这一步可选)

sudo service lightdm start



最后把cudnn的头文件拷贝到/usr/local/cuda-8.0/include下面,库文件拷贝到/usr/local/cuda-8.0/lib64下面

设置环境变量文件.bashrc或profile

export PATH=/usr/local/cuda-8.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:#LD_LIBRARY_PATH

更新环境变量

souce /etc/profile

安装完成。

接下来就可以安装tensorflow, caffe, mxnet深度学习框架了。




1
0
   

查看评论

相关内容

    暂无相关文章