深度学习框架【Megengine】的安装

---

前言

本文简介安装megengine深度学习框架，下边开始对深度学习框架megengine进行介绍和使用

---

一、MegEngine的前世今生

MegEngine 是基于计算图的深度神经网络学习框架。
MegEngine 是一个快速、可拓展、易于使用的深度学习框架，拥有以下三大关键特点：

• 训练推理一体：训练推理同一内核，模型结构、量化、前后处理、动态 shape 甚至求导均可 放入模型 进行推理，训练推理轻松对齐精度
• 超低硬件门槛：依靠算法优化各类关键资源占用，DTR 让训练显存占用量一键下降3倍，pushdown 内存分配算法让推理内存占用下降至极低水平
• 全平台高效推理：在 x86/Arm/CUDA/RoCM 各平台上均可体验到高性能且精度对齐的推理体验，更有 丰富的高阶用法可以优化性能、节省内存

---

二、MegEngine的安装

megengine团队为了更好更快的快速部署，因此不存在编译问题，直接pip安装即可，但是需要注意：
MegEngine 目前支持在以下环境安装 Python 包：
操作系统： Linux-64 位 / Windows-64 位 / MacOS-10.14 及其以上
Python 版本：3.6 到 3.9
其中 MacOS 只支持 Intel x86 CPU； 此外，MegEngine 也支持在很多其它平台上进行推理运算。
megengine自带了gpu的安装，因此只需要电脑安装了nvidia驱动即可使用megengine，不需要额外安装cuda和cudnn

三、建议安装cuda10.1

注意，megengine的gpu和cpu版是同一个安装包，直接输入命令安装即可：

python3 -m pip install megengine -f https://megengine.org.cn/whl/mge.html