Ubuntu 18.04 及 cuda深度学习环境安装

一，安装ubuntu 18.04 desktop

1.下载镜像

https://ubuntu.com/download/desktop

2.制作启动u盘

如果是Windows就使用
etcher
https://www.balena.io/etcher/
或者rufus
https://rufus.ie/
制作启动u盘

3.开始安装

正常安装，记得安装的时候选上第三方驱动之类的

4.启动系统

这个时候应该是无法启动，因为带Nvidia独显的本本显示驱动会有问题，本人测试的是hp zbook 15v G5，带一个4G显存的Quadro P600
所以在启动界面要先禁用Ubuntu自带的nouveau驱动，方法启动的时候在ubuntu选择框中不要按回车进入ubuntu，按e，然后在linux一行最后加上

modprobe.blacklist=nouveau nvidia-drm.modeset=1

然后按F10启动，启动之后永久禁用nouveau驱动

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

添加

blacklist nouveau
options nouveau modeset=0

保存后重新生成kernel initramfs,输入下面两句命令

$ sudo update-initramfs -u
$ sudo update-grub

重启电脑

 $ sudo reboot

重启后应该就能直接进入系统，然后各种常规更新全部更新

sudo apt update
sudo apt upgrade

然后安装Nvidia官方驱动，先查看一下驱动

ubuntu-drivers devices

安装推荐的驱动

sudo ubuntu-drivers autoinstall

注意这里安装结束后会有一个提示安装私有驱动需要添加一个认证到系统里，输入密码重启后会弹出一个选择框，选择刚刚生成的认证文件输入密码后私有驱动就被签名了，之后就不会再跳出这个框框了。

二，安装Cuda和Cudnn

1.安装cuda

Nvidia官方链接：link
然后安装

sudo sh cuda_10.1.xxx.xx_linux.run

按空格跳过申明，然后注意看提示，不需要安装驱动，刚刚装过了，然后在~/.bashrc中添加

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

** 这边注意如果添加了/usr/local/cuda-10.1到/usr/local/cuda的软连接，上面第一行就写/usr/local/cuda,如果安装了多个版本的cuda，可以根据情况直接写成/usr/local/cuda-10.1或者/usr/local/cuda-10.0 **

然后测试一下：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

输出没问题就往下

2.安装cudnn

下载对应的cudnn
链接：link
然后复制到对应的cuda文件夹里把权限改一改就行了

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 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

查看一下cudnn版本：

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

查看一下显卡使用：

nvidia-smi

然后就可以开始搞pytorch或者tensorflow了