MDK - Caffe¶
相关文档在GTIMDK_V1.0.1或对应版本MDK中可找到
Ubuntu 16.04环境配置¶
所需硬件和软件的要求¶
所需硬件设备:
Linux(Ubuntu 16.04, 64位版本)
Intel i5或更好的处理器(i7 首选)3.0GHz或更快的CPU
8GB或更多内存
第三方软件:
Python 2.7或3.5
OpenCV 3.2.0
GNU 编译器(最低要求g++版本4.9, 首选5.4)
Caffe 1.0
Cuda 8.0和CUDNN 6
GTI SDK (MDK TensorFlow v4.0.1需要GTI SDK 5.x)
环境搭建¶
具体步骤可参考MDK - SSD - 环境安装。
与MDK - SSD中不同的是使用的代码不同。此版本的MDK使用的源码是https://github.com/BVLC/caffe 中的v1.0源码，加上一些修改。具体修改在下文描述。
在caffe中添加GTI NPU的支持¶
根据下表中列出的说明更新现有的Caffe 1.0框架。所有必需的文件都位于caffe-installation目录中（。
如果替换caffe.proto后，因caffe.proto编译不通过请使用合并的方式修改caffe.proto或者使用GTIcaffeMDKv3.0.0.1中提供的caffe源码。
模型训练流程说明¶
具体步骤请参考GTI Caffe MDK User Guide v3.0.0.1_CN.pdf
注：相关资料及源码可从AI资料U盘中获取，如未购买请先购买，已购买过的但资料不是最新的可联系客服获取最新的资料。