Ubuntu16.04 +gtx 1070+CUDA 8.0+ +caffe环境配置过程

一、安装GTX1070驱动

先干掉Nouveau,再安装Linux Nvidia显卡驱动,也即是禁用系统默认的集成驱动

Ubuntu系统集成的显卡驱动程序是nouveau，它是第三方为NVIDIA开发的开源驱动，我们需要先将其屏蔽才能安装NVIDIA官方驱动。

用gedit打开
$sudo gedit /etc/modprobe.d/blacklist.conf

在该文件后添加以下几行：

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

驱动所在网址http://www.geforce.com/drivers，可以查询

按Ctrl+Alt+F1切换到控制台界面，输入用户名密码登录

 

1、打开终端，先删除旧的驱动：

sudo apt-get purge nvidia*


2禁用自带的 nouveau nvidia驱动 （important！）

创建一个文件通过命令 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进入图形界面；

如果安装后驱动程序工作不正常，使用下面的命令进行卸载：

sudo sh ~/NVIDIA-Linux-x86_64-367.44.run --uninstall

5


nvidia-smi

nvidia-settings

 

进入附加驱动，更新一下

安装完成后，可以在终端查看是否安装成功

nvidia-smi

 

二、安装CUDA8.0

在CUDA 8的下载地址：https://developer.nvidia.com/cuda-downloads，下载与你GPU对应版本的CUDA

 sudo service lightdm stop //关闭图形化界面

在终端相应目录下执行：

sudo sh cuda_8.0.27_linux.run

 

一路accept、Y、default即可

 sudo service lightdm start //重启图形化界面

安装完成后，可以在终端查看是否安装成功

nvcc -V

 

 

三下载和安装cuDNN

 

安装cuDNN非常简单

tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz

得到一个cuda文件夹，把其中的文件拷贝到系统相应目录

 

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cd /usr/local/cuda/lib64/
sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig

$tar -zxvf cudnn-8.0-linux-x64-v5.1-prod.tgz  

$cd cuda 

$sudo cp lib64/lib* /usr/local/cuda/lib64/  

$sudo cp include/cudnn.h /usr/local/cuda/include/

更新软连接：

$cd /usr/local/cuda/lib64/

$sudo chmod +r libcudnn.so.5.1.10

$sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5

$sudo ln -sf libcudnn.so.5 libcudnn.so

更新设置：

$sudo ldconfig

 四，设置变量环境

终端输入：

PATH=/usr/local/cuda/bin:$PATH

export PATH

sudo gedit/etc/profile

按i插入，将上面export两句加到末尾，esc退出编辑，然后按：，wq保存

sudo vim/etc/ld.so.conf.d/cuda.conf

插入下面语句

/usr/local/cuda/lib64

五，安装caffe

参见博客:深度学习Caffe平台安装——Ubuntu安装CPU模式的caffe

将CPU模式改变GPU即可，其实，不用修改配置文件即可。

注意：整个安装过程在ubuntu下，所需下载的文件太多，太大，所以安装进度较慢。

"libcudart.so.8.0 cannot open shared object file: No such file or directory"

解决办法是将一些文件复制到/usr/local/lib文件夹下：

#注意自己CUDA的版本号！

sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfig

sudo cp /usr/local/cuda-8.0/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfig

sudo cp /usr/local/cuda-8.0/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfig

报错

$/sbin/ldconfig.real: /usr/lib/nvidia-375/libEGL.so.1 不是符号连接

$/sbin/ldconfig.real: /usr/lib32/nvidia-375/libEGL.so.1 不是符号连接

原因：

系统找的是一个符号连接，而不是一个文件。这应该是个bug....

解决方法：

1.对这两个文件更名

2.重新建立符号连接

$sudo mv /usr/lib/nvidia-375/libEGL.so.1 /usr/lib/nvidia-375/libEGL.so.1.org

$sudo mv /usr/lib32/nvidia-375/libEGL.so.1 /usr/lib32/nvidia-375/libEGL.so.1.org

$sudo ln -s /usr/lib/nvidia-375/libEGL.so.375.39 /usr/lib/nvidia-375/libEGL.so.1

$sudo ln -s /usr/lib32/nvidia-375/libEGL.so.375.39 /usr/lib32/nvidia-375/libEGL.so.1

$sudo ldconfig