CUDA8.0目前仍然是深度学习领域最常用的CUDA版本。对于搭建深度学习开发环境而言,CUDA8.0+cuDNNv5.1是一个经典选择,是TensorFlow1.2的安装要求。笔者第一次在Ubuntu16.04上安装CUDA8.0和cuDNNv5.1,以作记录。
一、安装CUDA前的准备工作
装好显卡驱动是安装CUDA前必不可少的工作,前边的博文已经有介绍:https://blog.csdn.net/qq_30163461/article/details/80314630。此外,CUDA还需要一些库的支持,通过指令完成安装:
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
如果没有安装mesa-common-dev和freeglut3-dev,则会在CUDA安装结束后出现以下提示:
Installing the CUDA Toolkit in /usr/local/cuda-8.0 ...
Missing recommended library: libGLU.so
Missing recommended library: libX11.so
Missing recommended library: libXi.so
Missing recommended library: libXmu.so
Installing the CUDA Samples in /home/zhou
意思是缺少一些库。如果安装CUDA前忘记以上两条指令,那么在装完CUDA后看到该提示再安装这些库也可以。
二、run文件方式安装CUDA
在官网中选择适合自己系统的CUDA8.0 runfile文件,我下载的是cuda_8.0.61_375.26_linux.run文件。
2.1 初始安装流程
首先进入run文件所在目录,并通过
sudo sh cuda_8.0.61_375.26_linux.run
指令开始CUDA的安装。
执行该指令后首先会有一个特别特别长的EULA显示,只能狂按ENTER键继续。接下里是安装的交互过程,其中需要注意的是不要再安装显卡驱动!因为前面我们已经安装了最新的nvidia384,所以这里不要选择安装。其余的都直接默认或者选择是即可。交互界面如下:
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
Enter Toolkit Location
[ default is /usr/local/cuda-8.0 ]:
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y
Enter CUDA Samples Location
[ default is /home/zhou ]:
安装结束后会出现如下界面:
===========
= Summary =
===========
Driver: