Ubuntu下安装CAFFE-Digits

最新推荐文章于 2021-02-07 01:19:20 发布

Striker_V

最新推荐文章于 2021-02-07 01:19:20 发布

阅读量4.1k

点赞数

分类专栏： DeepLearning

本文链接：https://blog.csdn.net/striker_v/article/details/53096279

版权

DeepLearning 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

搞了一段时间的Caffe，一直用的是python接口，做可视化和训练都可以写成脚本来使用，偶然听人说可以使用Digits来调用Caffe的图形界面，一开始我是拒绝的，觉得使用终端调用Caffe显得我比较牛X。无奈技术总是要更新，就尝试安装了一下，结果发现网上的攻略已经过时了，NVIDIA封锁了部分资源的下载，安装时会有404错误。于是翻阅了官网最新攻略，给大家参考一下。

官网攻略地址：https://github.com/NVIDIA/DIGITS/blob/digits-4.0/docs/UbuntuInstall.md

1.确定你是Ubuntu系统（最好是Ubuntu14.04），并配置成功了CUDA和Caffe

2.打开终端，切换到一个可以暂时存储下载包的目录，比如/disk

cd /disk
3.获取Root权限

sudo -s

获取权限之后的操作是不可逆的，不需要再输入密码，操作要小心。

4.获取CUDA包

# Access to CUDA packages
CUDA_REPO_PKG=cuda-repo-ubuntu1404_7.5-18_amd64.deb
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/${CUDA_REPO_PKG} -O /tmp/${CUDA_REPO_PKG}
sudo dpkg -i /tmp/${CUDA_REPO_PKG}
rm -f /tmp/${CUDA_REPO_PKG}

5.获取机器学习包

# Access to Machine Learning packages
ML_REPO_PKG=nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/${ML_REPO_PKG} -O /tmp/${ML_REPO_PKG}
sudo dpkg -i /tmp/${ML_REPO_PKG}
rm -f /tmp/${ML_REPO_PKG}

6.获取安装源

# Download new list of packages
sudo apt-get update

7.开始安装Digits

sudo apt-get install digits

8.启动Digits试试

直接打开浏览器，输入IP地址：http://localhost/，就可以看到界面啦

Digits的安装目录为：/usr/share/digits

作为一个负责任的博主，再送大家一条攻略。安装完成之后输入localhost可以在浏览器中打开Digits，这是因为此时自动开启了Digits服务。那么关机再开机之后呢？你会神奇的发现在浏览器输入localhost之后连毛球都打不开。网上攻略说的很麻烦，各种配置，其实方法很简单，在上面的官方安装攻略中就有答案。

启动Digits服务：

% sudo start nvidia-digits-server

关闭Digits服务：

% sudo stop nvidia-digits-server

启动Digits服务之后在浏览器输入 localhost就可以打开Digits了。请叫我红领巾～

无语的是，Digits是使用启用服务器的方式，通过浏览器就可以使用。这就是说，只要有一台配置好Digits的机器A，开机后连上网，整个局域网之内的机器都可以使用浏览器调用Digits和Caffe。不论windows还是linux，不论配置如何，都可以使用机器A来训练自己的网络，只是在输入IP地址时把localhost改成机器A的IP地址即可（Ubuntu终端输入ifconfig -a查看IP地址，使用"eth0"那一栏的地址）。好一个跨平台。。。NVIDIA为了卖板子真是不遗余力，连这种逆天软件都开发，，，

关于Digits的使用这里就不赘述了，提供两个参考资料：

一个比较详细的参考博客：