zl程序教程

您现在的位置是:首页 >  其他

当前栏目

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

2023-09-27 14:21:00 时间


前言

本文简介安装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