NVIDIA DIGITS 安装使用教程

欢迎访问我的个人博客：zengzeyu.com

前言

DIGITS: Deep Learning GPU Training

System1，是由英伟达（NVIDIA）公司开发的第一个交互式深度学习GPU训练系统。目的在于整合现有的Deep

Learning开发工具，实现深度神经网络（Deep Neural

Network，DNN）设计、训练和可视化等任务变得简单化。DIGITS是基于浏览器的接口，因而通过实时的网络行为的可视

化，可以快速设计最优的DNN。DIGITS是开源软件，可在GitHub上找到，因而开发人员可以扩展和自定义DIGITS。

Github界面：https://github.com/NVIDIA/DIGITS/tree/master/docs

Ubuntu 14.04 安装DIGITS

DIGITS依赖Caffe和CUDA，首先要安装这两个。

安装教程原文参考：https://github.com/NVIDIA/DIGITS/blob/master/docs/BuildDigits.md

1. 安装条件

NVIDIA显卡 + 驱动 （废话）

然后在终端运行下面命令：

# For Ubuntu 14.04
CUDA_REPO_PKG=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb
ML_REPO_PKG=http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb

# Install repo packages
wget "$CUDA_REPO_PKG" -O /tmp/cuda-repo.deb && sudo dpkg -i /tmp/cuda-repo.deb && rm -f /tmp/cuda-repo.deb
wget "$ML_REPO_PKG" -O /tmp/ml-repo.deb && sudo dpkg -i /tmp/ml-repo.deb && rm -f /tmp/ml-repo.deb

# Download new list of packages
sudo apt-get update

2. 依赖环境

sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-pip python-scipy python-tk

3. 下载源码

# example location - can be customized
DIGITS_ROOT=~/digits
git clone https://github.com/NVIDIA/DIGITS.git $DIGITS_ROOT

4. 安装 Python 包

sudo pip install -r $DIGITS_ROOT/requirements.txt

启动DIGITS

运行：

./digits-devserver

打开浏览器，输入

http://localhost:5000/.

，就可看到界面。

其他帮助：

$ ./digits-devserver --help
usage: __main__.py [-h] [-p PORT] [-d] [--version]

DIGITS development server

optional arguments:
-h, --help            show this help message and exit
-p PORT, --port PORT  Port to run app on (default 5000)
-d, --debug           Run the application in debug mode (reloads when the
source changes and gives more detailed error messages)
--version             Print the version number and exit

用户手册

参考官方文档： https://github.com/NVIDIA/DIGITS/blob/master/docs/GettingStarted.md

另：Windows 安装教程

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

以上。