Ubuntu16.04安装配置Caffe教程(GPU版)

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