Ubuntu 16.04 安装 NVIDIA DIGITS

最新推荐文章于 2020-10-25 11:40:58 发布

原创最新推荐文章于 2020-10-25 11:40:58 发布 · 1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#cuda #digits #caffe

深度学习专栏收录该内容

3 篇文章

订阅专栏

本文介绍如何在Ubuntu 16.04环境下安装NVIDIA的DIGITS深度学习服务器。包括安装CUDA 8.0、配置Caffe 0.15及TensorFlow等关键组件的过程。

DIGITS是nvidia公司出品的开源可视化深度学习服务器，可以简单的通过界面操作完成数据准备、深度网络模型设计、数据训练和测试等工作。它支持多种深度学习基础平台，包括 caffe, tensorflow, mxnet等等，本文以 caffe 和 tensorflow为例，讲解如何在ubuntu 16.04下安装。

首先，安装基本支持库。

CUDA 8.0 + Caffe + tensorflow

1) CUDA 8.0
下载地址：https://developer.nvidia.com/cuda-downloads

建议下载 linux -> X86_64 -> ubuntu -> 16.04 -> deb(local)

之后

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

注意：部分人反馈安装使用deb文件会有问题，我没有遇到。如果有问题可以采取.run的本地可执行文件安装。

2) 安装 caffe
安装 caffe建议从 nvidia的官方提供的caffe安装：
https://github.com/NVIDIA/DIGITS/blob/master/docs/BuildCaffe.md
使用 BVLC的版本理论上也可以，但是我没有去确认。
注意 caffe安装需要用到不少的支持库，请确保在对应的版本上编译。比如

安装 glog ， gflags等，我采用了源代码自己编译的模式；
注意编译的时候要用 -fPIC选项，否则后续链接会有问题；
方法：
在cmake的时候改变一下CXXFLAGS

export CXXFLAGS=”-fPIC” && cmake .. && make VERBOSE=1

步骤：2.1）安装需要的库
sudo apt-get install –no-install-recommends build-essential cmake git gfortran libatlas-base-dev libboost-filesystem-dev libboost-python-dev libboost-system-dev libboost-thread-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libsnappy-dev python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-pydot python-scipy python-skimage python-sklearn

2.2）安装caffe 0.15
export CAFFE_ROOT=~/caffe
git clone https://github.com/NVIDIA/caffe.git $CAFFE_ROOT -b ‘caffe-0.15’

python库支持
sudo pip install -r $CAFFE_ROOT/python/requirements.txt
这个过程会稍微长一点，请耐心等待；
或者采用清华的源：
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

然后
cd $CAFFE_ROOT mkdir build cd build cmake .. make -j"$(nproc)"

3) 安装 digits

DIGITS_ROOT=~/digits
git clone https://github.com/NVIDIA/DIGITS.git $DIGITS_ROOT

之后
sudo pip install -r $DIGITS_ROOT/requirements.txt
安装额外支持

sudo pip install -e $DIGITS_ROOT

4）安装完成，可以启动digits

注意此时的 CAFFE_ROOT要指向你编译的caffe地址

启动：./digits-devserver

浏览器打开 localhost:5000

就出来了界面

后续会讲如何使用和开发digits