Ubuntu 14.04下CUDA8.0 + cuDNN v5 安装配置过程

一、硬件与环境

显卡:GTX 1080
系统:Ubuntu 14.04
CUDA:cuda_8.0.44_linux.run
cuDNN:cudnn-8.0-linux-x64-v5.1.tgz

注意:

  1. GTX1080显卡必须用CUDA 8.0版本。CUDA从此处下载。切记,千万不要下载 deb 包,否则后方无数坑在等着你。

    CUDA下载界面
  2. GTX1080显卡必须用cuDNN-8.0-V5.1版本,不然用 caffe 跑模型,用 CPU或GPU显卡跑精度正常,一旦开启cuDNN模式,精度(acc)立刻下降到 0.1 左右,loss 非常大。cuDNN在此处下载。下载需注册。最好注册一个账号,选择对应的版本,不要用网上其他教程给的现成的包,出问题的概率非常大。

二、安装:

注意:一定要按顺序!
注意:一定要按顺序!
注意:一定要按顺序!

1. 安装 Ubuntu 14.04

本文不关注。假设你已经将此系统做过稳定的开发环境,这不是全新安装后的系统,至少 git 等最常用包已经安装。缺什么装什么,apt-get 大法好。

2. 禁用 nouveau 驱动

打开终端,输入$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf,输入

blacklist nouveau
options nouveau modset=0

保存退出,执行$ sudo update-initramfs -u,禁用结束。重启电脑。

验证驱动是否禁用成功:

输入$ sudo lspci | grep nouveau,如果没有内容,则禁用成功。


什么都没有,则禁用成功

3. 安装CUDA 8.0

运行下载好的 run file (假设 cuda_8.0.44_linux.run 在家目录下)。
$ sudo sh cuda_8.0.44_linux.run
一路回车即可。
验证是否安装成功:
输入$ ls /dev/nvidia*,若生成 4 个左右 Nvidia 开头的文件(夹),说明此步安装成功。此时已经安装好显卡驱动和CUDA 8.0。输入$ nvidia-smi可查看显卡驱动和其他信息。


生成了五个 nvidia 开头的文件夹

显卡驱动版本信息

注意:网上其他教程由于年代久远,里面不建议安装此 CUDA 包中的显卡驱动。然而,这个包中的显卡驱动大版本为 367 ,非常新,可以用。

4. 配置环境变量

输入$ sudo gedit /etc/profile,打开 gedit ,最后两行输入

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

保存退出,环境变量配置完成。

5. 安装 cuDNN

你已经下载好 cudnn-8.0-linux-x64-v5.1.tgz,并将其放在家目录($ cd ~)下。
按顺序输入以下代码:

$ cd ~
$ sudo tar xvf cudnn-8.0-linux-x64-v5.1.tgz
$ cd cuda/include
$ sudo cp *.h /usr/local/include/
$ cd ../lib64
$ sudo cp lib* /usr/local/lib/
$ cd /usr/local/lib# sudo chmod +r libcudnn.so.5.1.5
$ sudo ln -sf libcudnn.so.5.1.5 libcudnn.so.5
$ sudo ln -sf libcudnn.so.5 libcudnn.so
$ sudo ldconfig

注意:libcudnn.so后面跟的数字可能和你下载的 cudnn 包小版本的不同而不同,去~/cuda/lib64下看一眼,相对应地进行修改。

转载自http://www.linuxidc.com/Linux/2017-01/139313.htm

http://www.linuxidc.com/Linux/2017-01/139313.htm

你可能感兴趣的:(深度学习)