默认安装的系统为Ubuntu 16.04,cuda 8.0。digits作为caffe的一个可视化界面,对于初学者十分友好,caffe一直为人所诟病的就是十分不友好的可视化界面,有时候利用caffe画出一条曲线图都是那么的不方便,更别提曲线的美观了。
第一步,先在home目录下创建digits文件夹,从github上下载digits项目,执行如下命令:
git clone https://github.com/NVIDIA/DIGITS.git digits
从github上下载源码时尽量不要使用sudo,这会牵扯到后面文件执行权限的问题,即使你在后面文件执行的时候加上sudo,也Yui有一系列的报错,所以最保险的方式不要加上sudo.
第二步,利用digits文件夹中的requirements.txt文件进行相关依赖的安装配置:
cd digits
sudo apt-get install graphviz gunicorn
for req in $(cat requirements.txt); do sudo pip install $req; done
利用pip安装的时候可能会报错,网速可能会比较慢,需要耐心安装几次。最后能顺利安装。
此时直接运行digits中的digits-devserver则会报如下的错误:
解决措施如下:
第三步,在digits下执行如下命令:
./digits-devserver
出现如下界面则表示安装成功:
第四步,在浏览器中输入http://localhost:5000/ 则能进入digits的可视化界面。