Ubuntu 16.04 安装 NVIDIA DIGITS

DIGITS是nvidia公司出品的开源可视化深度学习服务器，可以简单的通过界面操作完成数据准备、深度网络模型设计、数据训练和测试等工作。它支持多种深度学习基础平台，包括 caffe, tensorflow, mxnet等等，本文以 caffe 和 tensorflow为例，讲解如何在ubuntu 16.04下安装。

首先，安装基本支持库。

CUDA 8.0 + Caffe + tensorflow

1) CUDA 8.0

下载地址：https://developer.nvidia.com/cuda-downloads

建议下载 linux -> X86_64 -> ubuntu -> 16.04 -> deb(local)

之后

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

注意： 部分人反馈安装使用deb文件会有问题，我没有遇到。如果有问题可以采取.run的本地可执行文件安装。

2) 安装 caffe

安装 caffe建议从 nvidia的官方提供的caffe安装：

https://github.com/NVIDIA/DIGITS/blob/master/docs/BuildCaffe.md

使用 BVLC的 版本理论上也可以，但是我没有去确认。

注意 caffe安装需要用到不少的支持库，请确保在对应的版本上编译。比如

安装 glog ， gflags等，我采用了源代码自己编译的模式；

注意编译的时候要用 -fPIC选项，否则后续链接会有问题；

方法：

在cmake的时候改变一下CXXFLAGS

export CXXFLAGS=”-fPIC” && cmake .. && make VERBOSE=1

步骤：2.1）安装需要的库

sudo apt-get install –no-install-recommends build-essential cmake git gfortran libatlas-base-dev libboost-filesystem-dev libboost-python-dev libboost-system-dev libboost-thread-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libsnappy-dev python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-pydot python-scipy python-skimage python-sklearn

2.2）安装caffe 0.15

export CAFFE_ROOT=~/caffe

git clone https://github.com/NVIDIA/caffe.git $CAFFE_ROOT -b ‘caffe-0.15’

python库支持

sudo pip install -r $CAFFE_ROOT/python/requirements.txt

这个过程会稍微长一点，请耐心等待；

或者采用清华的源：

[global]

index-url = https://pypi.tuna.tsinghua.edu.cn/simple

然后

cd $CAFFE_ROOT

mkdir build

cd build

cmake ..

make -j"$(nproc)"

3) 安装 digits

DIGITS_ROOT=~/digits

git clone https://github.com/NVIDIA/DIGITS.git $DIGITS_ROOT

之后

sudo pip install -r $DIGITS_ROOT/requirements.txt

安装额外支持

sudo pip install -e $DIGITS_ROOT

4） 安装完成，可以启动digits

注意此时的 CAFFE_ROOT要指向你编译的caffe地址

启动：./digits-devserver

浏览器打开 localhost:5000

就出来了界面

后续会讲如何使用和开发digits