digits工具是caffe的非常給力的預處理工具,開源地址爲https://github.com/NVIDIA/DIGITS
安裝在Ubuntu16.0.4上安裝命令如下所示:
源代碼安裝方式
1.安裝依賴包:
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-protobuf python-scipy
2.git程序到本地
DIGITS_ROOT=~/digits git clone https://github.com/NVIDIA/DIGITS.git $DIGITS_ROOT
3.安裝軟件依賴
sudo pip install -r $DIGITS_ROOT/requirements.txt
4.支持插件(可選)
sudo pip install -e $DIGITS_ROOT
5.添加CAFFE_ROOT環境變量(假設caffe的目錄在/home/work/caffe)
export CAFFE_ROOT=/home/work/caffe6.啓動digits的服務
./digits-devserver
7.在瀏覽器中查看
http://localhost:5000/