NVIDIA DIGITS 安装教程

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/

Home page

猜你喜欢

转载自my.oschina.net/u/3773937/blog/1786278