MDK - TensorFlow

相关文档在GTIMDK_V1.0.1或对应版本MDK中可找到

Ubuntu 16.04环境配置

所需硬件和软件的要求

  • Linux(Ubuntu 16.04 , 64 位版本)

  • CUDA 9.0或更高版本

  • CUDNN 7.0或更高版本

  • Python 3.6

  • tensorflow-gpu 1.12

  • numpy 1.15

  • tqdm 4.28 (用于训练过程的可视化)

  • GTI SDK　(MDK TensorFlow v3.0.1.0需要GTI SDK 5.x)

pip 可用于安装任何缺少的 Python 依赖项。建议使用 Anaconda / Miniconda 或Docker 来管理软件包和环境。 需要 GTI SDK 5.0 及更高版本来评估生成的模型。

要了解有关 GTI SDK 的更多信息,请参阅 GTI SDK 用户手册。

环境搭建

请参考tensorflow安装教程.zip/tensorflow/ubuntu16.04+cuda9.0+tensorflow1.12安装教程.pdf

模型训练流程

具体步骤请参考GTI TensorFlow MDK User Guide_v2.0.0.1_CN.pdf或随包的文档。

注：相关资料及源码可从AI资料U盘中获取，如未购买请先购买，已购买过的但资料不是最新的可联系客服获取最新的资料。