旷视「天元」深度学习框架上手：概况、安装和初步体验

概况

天元（英文名：MegEngine）是旷视科技 3 月 25 日开源的深度学习框架，这一名字取自围棋棋盘中心点的名称，也有向 AlphaGo 致敬之意。听了旷视大佬们在发布会上的介绍，MegEngine 从 2014 年开始研发，作为旷视内部全员使用的框架，MegEngine 是驱动旷视在深度学习领域取得一系列成绩的核心动力。

趁没开学在家，对照官方文档快速上手了一下，这篇文章会记录一下对 MegEngine 初步的一些感受。

• 天元 MegEngine 官方网站：https://megengine.org.cn/
• GitHub repo：https://github.com/MegEngine/MegEngine
• 中文社区：https://discuss.megengine.org.cn/
• MegStudio：https://studio.brainpp.com/

安装

目前 MegEngine 的支持平台还比较单一，仅支持 Linux 环境下安装。对 Windows 用户，官方提示了可以使用 WSL 来运行 MegEngine，不过只支持 CPU 后端。

详细的环境要求是：

• 64 位、16.04 及以上版本的 Ubuntu
• Python 3.5+
• NVIDIA 驱动版本 418.x

旷视 MegStudio 平台提供了 MegEngine 0.3.1 + Python 3.8 的环境，可以很方便地开始使用 MegEngine。

除了 MegStudio，实测 Google Colab 平台也可以成功安装 MegEngine。

MegEngine 安装包本身集成了 CUDA 环境，因此不区分 CPU 和 GPU 版本。

通过 pip 安装 MegEngine：

pip3 install megengine -f https://megengine.org.cn/whl/mge.html

整个安装包 700MB+，下载的速度非常快，服务器在国内果然无惧速度问题。

import 一下来测试安装是否成功，官方对 MegEngine 的习惯性缩写是 mge：

import megengine as mge

对于要参与到 MegEngine 开发贡献的开发者，或者需要使用未进入 release 的功能，则需要从源码安装，可以参考 README 中的指引。

体验

MegStudio

MegStudio 是旷视开放的提供免费算力的在线深度学习开发平台。目前提供下面三种配置的环境，其中基础版的环境不限时长，高级版（CPU）和高级版（GPU）环境需要通过算力卡获得。目前算力卡是通过邀请用户的方式获得。

MegStudio 开发环境是基于 JupyterLab，环境关闭之后额外添加的文件会被销毁。使用体验上和 JupyterLab 基本没有区别，如果官方能默认支持一下代码补全就更好了。

友好的中文文档

对照 官方文档 上手的过程当中，很大的一个体会是 MegEngine 的文档做得很用心。

首先从语言上说，中文书写的文档对于国内的开发者来说无疑是很友好的。

官方文档目前分为基础学习和进阶学习两个部分，MegEngine 团队设计了一个循序渐进的入门指引帮助使用者熟悉 MegEngine。

MegEngine 的文档中穿插了对神经网络知识的简要讲解，比如下面是文档中介绍的 BP 过程。

文档代码中的注释也非常详细，介绍了代码释义，和 API 的细节，阅读起来会比较轻松。