以Ubuntu16.04环境下安装cuda9.0与cudnn8.0为例
安装ubuntu系统
#安装默认依赖项
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 libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install git cmake build-essential
屏蔽自带驱动
在终端下输入:
sudo gedit /etc/modprobe.d/blacklist.conf
在文件最后部分插入以下两行内容
blacklist nouveau
options nouveau modeset=0
更新系统
sudo update-initramfs -u
重启系统
验证nouveau是否已禁用,没有信息显示,说明nouveau已被禁用
lsmod | grep nouveau
安装英伟达驱动
按ctrl+alt+f1
进入文本模式
关闭图形输出
sudo service lightdm stop
cd到驱动所在的文件目录
修改驱动文件读取权限
chmod 777 NVIDIA-Linux-x86_64-430.40.run
执行.run文件,记得–no-opengl-files,即只安装驱动,不安装OpenGL文件
sudo ./NVIDIA-Linux-x86_64-430.40.run –no-opengl-files
根据默认选项一路安装到结束
查看显卡信息
nvidia-smi
安装CUDA
再次按ctrl+alt+f1进入文本模式
关闭图形输出
sudo service lightdm stop
cd到驱动所在的文件目录
修改驱动文件读取权限
chmod 777 cuda_9.0.176_384.81_linux.run
执行.run文件
sudo ./cuda_9.0.176_384.81_linux.run
按空格阅读完用户协议
进入安装选项,具体选项如下,切勿安装OpenGL
![](https://zhenhuizhang.github.io/post-images/1577515813808.png)
输入nvidia-smi,查看显卡信息
重启图形化界面
sudo service lightdm start
重启电脑
测试CUDA是否安装成功,执行以下命令:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
如果显示一些关于GPU的信息,则说明安装成功。
安装cudnn
下载对应版本的cudnn软件,下载cuDNN时也一定要注意与CUDA版本的适配性,若文件后缀错乱改成tgz
将下载的源文件放在/usr/local/目录下
sudo cp cudnn-8.0-linux-x64-v6.0.tgz /usr/local
cd /usr/local
文件解压缩
sudo tar -xvzf cudnn-8.0-linux-x64-v6.0.tgz
修改文件权限
sudo chmod 777 /usr/local/cuda/include/cudnn.h
sudo chmod 777/usr/local/cuda/lib64/libcudnn*
拷贝解压缩文件到对应的cuda目录
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/ -d
安装三个库文件
sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb
sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb
sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb