Ubuntu安装配置Caffe教程(GPU版)
By yggao
此教程主要借鉴自:
GPU版:http://www.linuxidc.com/Linux/2016-12/138870.htm
CPU版:http://www.linuxidc.com/Linux/2016-09/135034.htm
1 电脑配置及软件说明
1) Ubuntu 16.04 x64系统,GPU:Quadro M4000(提示:在linux下可以通过命令lspic | grep –i vga 查看)
2) Caffe:Convolutional Architecture for Fast FeatureEmbedding
3) NVIDIA显卡驱动(不同的GPU对应不同的版本)
4) CUDA 9.0
5) cuDNN v7.0.5 for CUDA 9.0
6) OpenCV 3.2.0
2 安装NVIDIA GPU驱动
2.1 查询GPU驱动
首先去NVIDIA官网http://www.nvidia.com/Download/index.aspx?lang=en-us查看适合自己显卡的驱动并下载。如下图1,图2所示。
图1. 填写GPU和系统信息
图2. 搜索结果
本文中显卡型号是Quadro M4000,系统linux 64-bit,按照要求点击serach出现图2所示搜索结果,点击下载。本文中下载的驱动文件是:NVIDIA-Linux-x86_64-384.111.run。
注意:我们要把这个文件放到英文目录(e.g. /home)下,原因是下面我们要切换到文字界面下,如果放到中文目录下,我们没有办法进入这个目录(没有中文输入法,且中文全部是乱码)。
2.2 安装GPU驱动
2.2.1 屏蔽自带显卡驱动
1) 在终端下输入: sudo gedit /etc/modprobe.d/b