DIGITS(Deep learning GPU Training System)是一款用于训练深度学习模型的Web应用程序。目前支持的框架是:Caffe,Torch和Tensorflow。
您可以在这里下载源码:https://gitee.com/NVIDIA_Developer_Community/DIGITS
一、安装环境和背景
DIGITS支持基础训练框架包括Caffe,TensorFlow,Torch,本文以Caffe为例。
DIGITS目前可以支持Ubuntu 14.04, Ubuntu 16.04,本文就不再赘述系统安装的基本操作,如果需要可以再网上找到大量的教程。
环境配置:
- Ubuntu 16.04
- CUDA 9.0
- CUDNN 7.0
- Caffe-0.15
二、详细安装过程
1.cuda安装(如果您已经安装好了,就不用执行这步):
# For 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
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
注意如果您没有安装CAFFE,您可以在这里找到详细教程:
https://github.com/NVIDIA/DIGITS/blob/digits-6.0/docs/BuildCaffe.md
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
5.最后,安装DIGITS:
sudo pip install -e $DIGITS_ROOT
安装完成之后,您可以按照下面的命令启动DIGITS:
./digits-devserver
然后随便打开一个浏览器(您可以在本机用浏览器访问,也可以利用在同一网络内的任何一台电脑的浏览器访问),访问:http://Your Digits Server IP:5000/