Dragon深度框架的ubuntu环境搭建

2018-03-15

Dragon: A Computation Graph Virtual Machine Based Deep Learning Framework

github Dragon code
最近学习了深度框架Dragon，代码在github开源的，目前可能用的人不多，但是毕竟是深度新框架，值得一学习，在这里我根据文档给出安装步骤。借用Dragon原作者的一句话：

TensorFlow并不是最好的框架，MXNet也不是，最好的框架是自己用的舒服的框架。

安装

本人只进行了linux环境的搭建，windows环境搭建需要参考github。

1、clone dragon

1	git clone https://github.com/seetaresearch/Dragon.git

将clone后的Dragon文件夹改名为REPO_ROOT；REPO_ROOT、Dragon改名为DRAGON_ROOT。

2、Setup Python Development Environment

1	sudo apt-get install libpython-dev

当然，也使用的是Anaconda（建议使用2.x版本）

3、Setup C++ Development Environment

1
2
3

sudo apt-get install libprotobuf-dev
sudo apt-get install protobuf-compiler
sudo apt-get install libopenblas-dev

4、Install Python Requirements

1 2	cd $DRAGON_ROOT/python pip install -r requirements.txt

5、Configure DRAGON_ROOT/CMakeLists.txt

1 2	Select optional libraries [PYTHON / BLAS / SSE] (Optional) Set PYTHON_EXECUTABLE if you want to use specific Python (e.g. Anaconda)

最好开启PYTHON、 BLAS、 SSE选项，如果有CUDA可以选择开启。

6、Install CMake

1	sudo apt-get install cmake

7、Make

cd $DRAGON_ROOT
mkdir build
cd build
cmake ..
make install -j16

这一步的时候如果用seetatech版本报错cuda_device.h,可能代码语顺的原因，所以用的第一套Dragon版本就没有报错。如果需要修改DRAGON_ROOT/CMakeLists.txt，这里需要进行make clean，重新进行编译make instal。

8、Install Dragon

1
2
cd $DRAGON_ROOT/python
python setup.py install



如果报错说没有权限，可以进行如下命令操作：

1
2
cd $DRAGON_ROOT/python
python setup.py install --user



本人用的是：

1
sudo python setup.py install


当然这是在有sudo权限的基础上能够进行，可能一些企业没有sudo命令权限。