Ubuntu18-CUDA9.0+cuDNN7.1.4
2018-08-17 本文已影响0人
Joy_chan
一、安装好系统后先做如下操作：
1、换源：阿里的
2、下载CUDA、CUDXNN（驱动不用专门下了）
二、验证系统是否安装了kernel header和 package development
uname -r
sudo apt-get install linux-headers-$(uname -r)
2、安装 gcc
sudo apt-get build-dep gcc-6
sudo apt-get install build-essential
ls /usr/bin/gcc*
ls /usr/bin/g++*
2.1、由于已有gcc-7的版本，需要将某个版本加入gcc候选中，最后的数字是优先级，我自己是直接设为50，指令如下：
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 50
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 40
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/g++-6 50
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/g++-7 40
2.2、完成上面的操作之后，我们就可以通过下面的指令来选择不同的gcc和g++的版本了，（注意使用之前确保至少有两个gcc或者g++的版本进行了第2.2步的操作）：
sudo update-alternatives --config gcc
3、屏蔽开源驱动nouveau
方法一：在运行nvidia驱动安装程序时，会提示是否需要屏蔽，选yes即可（需要重启）
方法二：编辑blacklist.conf，在终端输入：
sudo gedit /etc/modprobe.d/blacklist.conf
在文档最后添加如下内容（不需要重启）
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
4、禁止自带的nouveau nvidia驱动
打开配置文件，在终端输入：
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
在文档最后填写禁止配置的内容：
blacklist nouveau
options nouveau modeset=0
更新配置文件，在终端输入：
sudo update-initramfs -u
5、安装CUDA、cuDNN
5.1、第3步禁用nouveau后不要安装驱动，重启后进入字符界面
5.2、安装CUDA
是否安装驱动？答：Y
是否XXXXXXX？答：N
是否启动X服务？答：N
后面的操作跟以前一样
4.3、安装完成后重启
4.4、验证是否安装成功，
在终端输入：
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
然后编译并运行示例，运行成功则表示安装成功。
如果安装成功会输出：
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.2, CUDA Runtime Version = 9.2, NumDevs = 1
Result = PASS
4.5、添加环境变量：
sudo gedit ~/.bashrc
在文本最后添加以下内容：
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
4.6最后执行
source ~/.bashrc
4.7、安装cudnn
终端进入cudnn所在文件夹，输入以下命令：
tar -xzvf cudnn-9.1-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
4.8、建立软连接
终端输入
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.1.4 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
4.9、设置环境变量，终端输入
sudo gedit /etc/profile
在文档末尾加入
PATH=/usr/local/cuda/bin:$PATH
export PATH
保存后，创建链接文件
sudo gedit /etc/ld.so.conf.d/cuda.conf
增加下面内容
/usr/local/cuda/lib64
最后在终端输入
sudo ldconfig
使链接生效
5、安装依赖/git/vim
sudo apt-get install build-essential #必要的编译工具依赖
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev
protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev git vim
