安装完caffe,继续安装Digits 。
DIGITS 是英伟达为普及深度学习开发的图形化操作界面,简单易用,旨在帮助初学者跨越入门障碍,迅速上手。因此,DLI 的入门培训均建议学员从 DIGITS 起步。 说白了, DIGITS 就是一个新手工具。
一、安装依赖项
sudo apt install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-scipy python-tk
二、下载安装DIGITS
git clone https://github.com/NVIDIA/Digits
sudo -H pip install -r Digits/requirements.txt
sudo -H pip install -e Digits/
三、配置环境变量
sudo nano ~/.bashrc
export CAFFE_ROOT=刚刚安装caffe的路径
source ~/.bashrc
四、运行digits-devserver
出现如下错误:
解决方法:
sudo -H pip install flask-wtf==0.14.3 -i https://pypi.doubanio.com/simple
flask-wtf 需要安装新版本