English | 中文
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架。
注意: MegEngine 现在仅支持 Linux 平台安装,以及 Python3.5 及以上的版本(不支持 Python2 )。对于 Windows 10 用户,可以通过安装 WSL(Windows Subsystem for Linux) 进行体验。
通过 pip 安装的命令如下:
pip3 install megengine -f https://megengine.org.cn/whl/mge.html
大多数编译 MegEngine 的依赖位于 third_party
目录,可以通过以下命令自动安装:
$ ./third_party/prepare.sh
$ ./third_party/install-mkl.sh
但是有一些依赖需要手动安装:
MegEngine 遵循“源外构建”(Out-of-Source Build)原则,并且使用静态编译方式。编译的具体流程如下:
创建用于编译的目录:
mkdir -p build
cd build
使用 CMake
生成编译配置:
生成支持 CUDA 环境的配置:
cmake .. -DMGE_WITH_TEST=ON
生成仅支持 CPU 环境的配置,使用 -DMGE_WITH_CUDA=OFF
选项:
cmake .. -DMGE_WITH_CUDA=OFF -DMGE_WITH_TEST=ON
生成仅用于 C++ 环境部署的配置,使用 -DMGE_INFERENCE_ONLY=ON
,并可用 -DMGE_WITH_TEST=OFF
关闭测试:
cmake .. -DMGE_INFERENCE_ONLY=ON -DMGE_WITH_TEST=OFF
可以使用 -DCMAKE_INSTALL_PREFIX=YOUR_PATH
指定具体安装目录。
开始编译:
make -j$(nproc)
[可选] 如果需要用于部署,可以安装 MegEngine 的 C++ 库:
make install
以下是其它常用编译选项:
MGE_ARCH
指定编译的目标平台(默认自动检测当前平台)MGE_WITH_DISTRIBUTED
是否开启多机分布式支持(默认开启)MGE_WITH_PYTHON_MODULE
是否编译生成 Python 模块(默认开启)MGE_BLAS
选择 BLAS 的后端实现,可以是 MKL
或 OpenBLAS
(默认 MKL
)MGE_CUDA_GENCODE
指定提供给 nvcc
的 -gencode
选项(默认不指定)MGE_DISABLE_FLOAT16
是否不提供 float16
类型支持(默认关闭)MGE_ENABLE_EXCEPTIONS
是否开启 C++ 报错支持(默认开启)MGE_ENABLE_LOGGING
是否开启 MegEngine 日志信息(默认自动检测)更多选项可以通过以下命令查看:
cd build
cmake -LAH .. 2>/dev/null| grep -B 1 'MGE_' | less
我们相信我们能够搭建一个开放友善的开源社区环境,用人工智能造福人类。
MegEngine 使用 Apache License, Version 2.0
Copyright (c) 2014-2020 Megvii Inc. All rights reserved.
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。
1. 开源生态
2. 协作、人、软件
3. 评估模型