一、 准备工作
1. 首先下载英伟达驱动,下载地址。根据自己的系统以及显卡型号选择驱动,笔者下载的是NVIDIA-Linux-x86_64-367.57.run
2. 下载CUDA,下载地址。笔者选择的是CUDA8.0,目前1070和1080好像只支持8.0。其他显卡的也可以选择CUDA7.5.
3. 下载cudnn,下载地址。要填写一个小问卷,很快的。根据自己选择的CUDA版本选择cudnn版本,笔者选择的是cudnn v5.1 for CUDA8.0
4. 下载tensorflow,下载地址。这里我们选择安装编译好的二进制包,这是最简单的方法。在下载页面如下图位置,选择Linux-GPU:Python3.5
二、 安装显卡驱动、CUDA、cudnn
1. 安装显卡驱动
如果是新安装的Ubuntu系统,那么直接安装按以下步骤安装即可。如果曾经装过显卡驱动,请先卸载旧版驱动。
a) Ctrl+Alt+F1进tty,如果黑屏可以修改grub文件。具体做做法如下
sudo gedit /etc/default/grub 用gedit打开grub文件
将GRUB_CMDLINE_LINUX_DEFAULT的值修改为 quiet splash nomodeset
sudo update-grub 更新grub
重启电脑(可能会发现系统分辨率变了,不要紧,安装好驱动就会恢复正常)
b) 进入tty之后,输入用户名和密码。
c) sudo service lightdm stop 关闭图形界面
sudo chmod 755 NVIDIA-Linux-x86_64-367.57.run获取权限,根据自己驱动名修改
sudo ./NVIDIA-Linux-x86_64-367.57.run 如果提示pre-install失败可以忽略
sudo service lightdm start 开启图形界面
d)测试一下,Ctrl+Alt+t打开terminal
cat /proc/driver/nvidia/version
如果出现下面的界面
那么说明驱动安装成功了,可以看到笔者安装的版本是367.57
2. 安装CUDA
a) 首先安装相关的依赖库
sudo apt-get install freeglut3-dev
sudo apt-get install build-essential
sudo apt-get install libx11-dev
sudo apt-get install libxmu-dev
sudo apt-get install libxi-dev
sudo apt-get install libglu1-mesa
sudo apt-get install libglu1-mesa-dev
b) 切换到放cuda安装包的文件夹
sudo sh cuda_8.0.44_linux.run --override
c) 这时候会出现一大堆协议,一直按space就可以,最后accept
d) 需要特别注意:安装过程中问你要不要Install NVIDIA ~~ drive一定要选n,不然之前装的显卡驱动就白装了。这里主要是因为cuda提示安装的驱动版本比较低。
e) 其他都选yes,默认文件夹不用去改它。
f) 安装完毕之后设置一下环境变量。
将下面两行写到~/.bashrc文件中(sudo gedit ~/.bashrc)
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda8.0-/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
g) 测试一下,切换到cuda_sample文件夹
cd ~/NVIDIA_CUDA-8.0_Samples 切换到sample文件夹
make make一下
cd bin/x86_64/linux/release 切换到release目录
./deviceQuery 运行一下实例
如果出现下面的图,说明安装成功了。
3. 安装cudnn
a) cd DL 切换目录,根据实际情况更改
tar xvzf cudnn-8.0-linux-x64-v5.1.tgz 解压
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 /usr/local/cuda/lib64/libcudnn*
这样子就可以了。
4. 安装anacond
anaconda集成了python的100多个库,比较方便。
先去官网下载,下载地址。
下载好后,切换到下载目录
bash Anaconda3-4.2.0-Linux-x86_64.sh
中间会问你要不要修改环境变量,选y
安装完成记得执行一下 source ~/.bashrc 才会生效
测试一下,在命令行输入python,出现下图说明安装正确。
5. 安装tensorflow
a) 首先执行下面两句,开启GPU加速
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
b) pip install tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
这就安装好了。你没看错。如果是自己折腾从源码编译不要太麻烦
c) 测试一下
cd anaconda3/lib/python3.5/site-packages/tensorflow/models/image/mnist
python convolutional.py
5.4ms,比cpu不要快太多。就这么愉快的结束了。真正的学习过程才刚刚开始~