前言

前一阵重做系统把之前的安装教程的博客给弄丢了,现在重写一份做记录好了。目前老环境还是要CUDA Tookit 8(以下简称CUDA8),因此目前先不安装CUDA最新的CUDA9了。

本教程是在Ubuntu 16.04.3 LTS上进行的,其他版本的Ubuntu系统请自行实验,目前在Ubuntu 17.10系统上实验成功。

当前假设Ubuntu系统已经安装完成,目前直接进行CUDA8的安装。

基础准备

基础环境安装

安装CUDA8需要基础的编译环境,需要检测下系统上是否安装了gccg++。命令如下:

1
2
$ gcc -v
$ g++ -v

如果任意一个出现版本信息就代表安装过了。如果都没有出现版本信息,则请采用如下命令安装:

1
$ sudo apt install build-essential

安装完成即可,网上有说CUDA8不支持g++ 5.0以上的版本,目前本人没有遇上这个问题。

附加环境安装

有些人在上述基础环境下安装完CUDA8后会出现Missing recommended library: libGLU.so的提示。如果不确定当前环境想提前避免这个问题,请安装如下包:

1
sudo apt install libglu1-mesa libxi-dev libxmu-dev libglu1-mesa-dev

至此,基础环境都安装完了。

NVIDIA驱动安装

个人强烈建议先装NVIDIA驱动,因为CUDA8自带的驱动实在是容易出问题。驱动安装详见「Ubuntu系统NVIDIA显卡驱动安装」。

CUDA8安装配置

CUDA8下载

先从官网下载CUDA的驱动。目前CUDA的最新版本是CUDA9,要下老版本的话请采用如下官方网址,下载Base InstallerPatch 2

个人建议先将文件都下载到家目录(~)底下,因为如果系统是中文环境的话,后续安装可能会出现文件夹名乱码的情况。安装完成后可以移动到自己想移动的位置。

CUDA8安装

是否进入tty1环境看个人,本人之前各种重装驱动给吓怕了,不确定图形界面是否影响到了,因此直接进入了tty1环境进行CUDA8的安装。关闭图形界面的步骤如下:

  1. 首先按住Ctrl+Alt+F1进入tty1
  2. 输入用户名和密码
  3. 执行sudo service lightdm stop命令关闭图形界面。

然后在安装文件所在目录下执行如下命令安装:

1
2
$ sudo chmod 755 (CUDA Install File)
$ sudo ./(CUDA Install File)

其中CUDA Install File是个人CUDA安装文件的名字(包括文件后缀)。

安装关键过程如下(先后顺序记不清了):

  1. 安装过程中先阅读完一大串协议,按住d往下(这样跳得很快),直到最后。然后输入accept。
  2. 会询问是否安装NVIDIA DRIVER,输入n
  3. 询问是否安装OPENGL时输入n。(记得有这个问题的,这里是个大坑!!)
  4. 其他默认选择y或者空着(就是直接按回车)就行。

到最后可能会出现INCOMPLETE INSTALL,这里不用管。这是因为你没装它的驱动而已。如果出现Missing recommended library: libGLU.so的提示,请参考附加环境安装

等一切都安装好后重启即可。

补丁安装

是否进入tty1环境依旧看个人,具体参考上一节开头。

安装时执行如下命令安装:

1
2
$ sudo chmod 755 (CUDA Patch File)
$ sudo ./(CUDA Patch File)

其中CUDA Patch File是之前下载的CUDA8的补丁文件Patch 2。安装过程和前一节类似,基本上直接一路默认就行。

等一切安装完了后重启即可。

环境设置

个人习惯在/etc/profile.d下设置环境变量。先进入该文件夹,然后执行如下命令:

1
2
$ sudo touch cuda.sh
$ sudo vim cuda.sh

这里vim可以换成gedit或者其他熟悉的编辑器。如果不熟悉vim的同学无意间进去了,输入:q退出vim

在编辑器中输入如下文字:

1
2
3
export CUDA_HOME=/usr/local/cuda-8.0
export PATH=$PATH:/usr/local/cuda-8.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64

然后保存重启即可。

cuDNN安装

cuDNN是个GPU加速库,能为深度学习网络的计算加速。在官网下载。下载前可能需要先注册NVIDIA DEVELOPER。注册过程很简单,注册完成后选择支持CUDA8.0cuDNN下载即可。

在下载目录解压后进入cuda文件夹,这里会见到includelib64两个文件夹。这里建议在命令行下执行,因为会用到sudo进行暂时的root权限申请。输入如下命令进行安装:

1
2
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64

然后进入/usr/local/cuda/lib64文件夹中,执行如下命令:

1
2
3
4
$ sudo rm libcudnn.so libcudnn.so.6
$ sudo ln -s libcudnn.so.6.0.20 libcudnn.so.6
$ sudo ln -s libcudnn.so.6 libcudnn.so
$ sudo ldconfig

这里.6.6.0.20是下载的cuDNN的版本号,请依据个人下载的实际版本进行修改。

至此,cuDNN已经安装完成。

结尾

至此,整个CUDA Tookit 8的所有安装过程就完成了。