[Cuda][Linux]ubuntu环境下cuda安装教程

以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

输入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

到此,在Ubuntu下的cuda、cudnn环境安装完成