NVIDIA DIGITS深度学习平台源码编译指南

NVIDIA DIGITS深度学习平台源码编译指南

DIGITS Deep Learning GPU Training System 项目地址: https://gitcode.com/gh_mirrors/di/DIGITS

前言

NVIDIA DIGITS（Deep Learning GPU Training System）是一个基于Web的交互式深度学习训练系统，它简化了深度学习模型的开发流程。本文将详细介绍如何在Ubuntu系统上从源码构建DIGITS平台，帮助开发者搭建自己的深度学习开发环境。

系统要求

硬件要求

• NVIDIA显卡（需支持CUDA）
• 推荐4GB以上显存

软件要求

• Ubuntu 14.04或16.04（其他版本可能兼容但不保证）
• NVIDIA显卡驱动（需先安装）

环境准备

1. 配置软件源

首先需要添加NVIDIA的CUDA和机器学习软件源：

# 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

# Ubuntu 16.04
CUDA_REPO_PKG=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
ML_REPO_PKG=http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb

# 安装软件源包
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

# 更新软件包列表
sudo apt-get update

2. 安装基础依赖

DIGITS运行需要以下基础软件包：

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

深度学习框架支持

DIGITS支持多种深度学习框架作为后端，以下是推荐安装的框架：

1. Caffe（必需）

Caffe是DIGITS的核心依赖，必须安装。安装方法请参考Caffe官方文档。

2. Torch7（推荐）

Torch7提供了Lua接口的深度学习支持，安装方法请参考Torch官方文档。

3. TensorFlow（推荐）

Google开发的TensorFlow也是可选后端之一，安装方法请参考TensorFlow官方文档。

获取DIGITS源码

建议将DIGITS克隆到用户主目录下：

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

安装Python依赖

DIGITS需要多个Python包支持，可以通过以下命令安装：

sudo pip install -r $DIGITS_ROOT/requirements.txt

可选插件支持

如果需要使用数据加载和可视化插件，需要以开发模式安装DIGITS：

sudo pip install -e $DIGITS_ROOT

启动DIGITS服务

完成所有安装后，可以通过以下命令启动开发服务器：

cd $DIGITS_ROOT
./digits-devserver

默认情况下，服务会运行在http://localhost:5000/。启动参数说明：

• -p/--port: 指定服务端口（默认5000）
• -d/--debug: 启用调试模式（自动重载代码并提供详细错误信息）
• --version: 显示版本信息

常见问题排查

如果遇到配置问题，可以参考DIGITS的配置文件说明进行调整。大多数配置项都有合理的默认值，特殊情况下才需要自定义配置。

进阶开发

对于想要参与DIGITS开发的用户，建议阅读开发环境设置指南，了解如何配置完整的开发环境。

结语

通过以上步骤，您已经成功搭建了NVIDIA DIGITS深度学习平台。接下来可以开始探索DIGITS提供的各种深度学习功能，如图像分类、目标检测等任务。DIGITS的Web界面使得深度学习模型的训练和评估变得直观简单，大大降低了深度学习的入门门槛。

DIGITS Deep Learning GPU Training System 项目地址: https://gitcode.com/gh_mirrors/di/DIGITS