一、硬件与环境
显卡:GTX 1080
系统:Ubuntu 14.04
CUDA:cuda_8.0.44_linux.run
cuDNN:cudnn-8.0-linux-x64-v5.1.tgz
注意:
- GTX1080显卡必须用CUDA 8.0版本。CUDA从此处下载。切记,千万不要下载 deb 包,否则后方无数坑在等着你。
CUDA下载界面 - 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
,如果没有内容,则禁用成功。
![Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe 安装配置](https://i-blog.csdnimg.cn/blog_migrate/a58b6f08e53b0a10f4bd0ed151f3997b.png)
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
可查看显卡驱动和其他信息。
![Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe 安装配置](https://i-blog.csdnimg.cn/blog_migrate/f6802878b28362bd8ba00bc1dc43f659.png)
![Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe 安装配置](https://i-blog.csdnimg.cn/blog_migrate/c8be9fffd3fafc875fb15b37676f7789.png)
注意:网上其他教程由于年代久远,里面不建议安装此 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