ubuntu16.04 深度学习系统环境搭建记录

一.安装Nvidia显卡驱动程序

1.卸载之前的Nvidia显卡驱动

 $ sudo apt-get remove --purge nvidia*

2.编辑/etc/modprobe.d/blacklist.conf文件

$ sudo gedit /etc/modprobe.d/blacklist.conf

(1)向文件中(结尾处)添加如下内容：

blacklist nouveau
options nouveau modeset=0

(2)执行如下命令：

$ sudo update-initramfs -u

(3)执行重启命令，重启服务器

$ sudo reboot

(4) 重启后，按 “Ctrl+Alt+F1” 进入命令行模式

$ sudo service lightdm stop

(5)进入NVIDIA-Linux-xxx_64-xxx.xxx.run文件所在目录，执行命令

$ chmod +x NVIDIA-Linux-xxx_64-xxx.xxx.run
$ sudo ./NVIDIA-Linux-xxx_64-xxx.xxx.run --no-opengl-files

(6)执行重启命令

$ sudo reboot

(7)在终端输入命令

$ nvidia-smi

二.安装CUDA

1.卸载CUDA：

$ sudo /usr/local/cuda-x.0/bin/uninstall_cuda_x.x.pl  # x.x为自己之前的cuda版本号，下面一行代码类似

2.删除残余文件：

sudo rm -rf /usr/local/cuda-x.x/  

3.安装CUDA:

进入CUDA安装文件的下载目录，为安装文件添加可执行属性：

$ chmod +x cuda_10.0.130_410.48_linux.run  # 注意自己的cuda版本号

4.执行安装包，开始安装:

$ ./cuda_10.0.130_410.48_linux.run

安装过程中，会出现安装说明，空格键往下翻页，直接按 Ctrl+C 到文末，同意条款，输入 accept

5.出现四个需要回答yes/no 的问题

第一个，Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
不需要安装驱动，输入 no；
第二个，nstall the CUDA 10.0 Toolkit?
必须安装Toolkit,输入 yes；
第三个，Do you want to install a symbolic link at /usr/local/cuda?
同意建立软链接，输入 yes；
第四个，Install the CUDA 10.0 Samples?
是否安装测试
输入 yes/no 都可以；

第二个问题后会提示是否使用默认路径，直接回车通过

开始安装…

6.配置环境变量

安装完成后，需要配置环境变量，在终端输入

$ sudo gedit ~/.bashrc

打开文件后，在结尾添加下面几行：

export CUDA_HOME=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
export PATH=${CUDA_HOME}/bin:${PATH}

使文件生效：

$ source ~/.bashrc

在 /etc/profile中同样添加写入上面三行声明

$ sudo gedit /etc/profile

7.验证CUDA的版本

$ nvcc -V

三.安装cuDNN

在下载目录下解压文件：

$ tar -axvf cudnn-10.0-linux-x64-v7.tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

进入cuda的安装路径

$ sudo chmod +r libcudnn.so.7.0.5
$ sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
$ sudo ln -sf libcudnn.so.7 libcudnn.so
$ sudo ldconfig     # 查看cuda目录中的版本号

查看cuDNN版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

结束。