English | 中文
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架。
注意: MegEngine 现在支持 Linux-64bit/Windows-64bit/macos-10.14及其以上 (MacOS只支持cpu) 平台安装，支持Python3.5 到 Python3.8。对于 Windows 10 用户，可以通过安装 Windows Subsystem for Linux (WSL) 进行体验，同时我们也原生支持Windows。
通过 pip 安装的命令如下：
python3 -m pip install megengine -f https://megengine.org.cn/whl/mge.html大多数编译 MegEngine 的依赖位于 third_party 目录，可以通过以下命令自动安装：
$ ./third_party/prepare.sh
$ ./third_party/install-mkl.sh但是有一些依赖需要手动安装：
- CUDA(>=10.1), cuDNN(>=7.6) ，如果需要编译支持 CUDA 的版本。
- TensorRT(>=5.1.5) ，如果需要编译支持 TensorRT 的版本。
- LLVM/Clang(>=6.0) ，如果需要编译支持 Halide JIT 的版本（默认开启）。
- Python(>=3.5), Numpy, SWIG(>=3.0) ，如果需要编译生成 Python 模块。
MegEngine使用CMake作为构建工具。我们提供以下脚本来帮助编译:
-
host_build.sh 用于本地编译。 参数 -h 可用于查询脚本支持的参数:
scripts/cmake-build/host_build.sh -h -
cross_build_android_arm_inference.sh 用于ARM-安卓交叉编译。 参数 -h 可用于查询脚本支持的参数:
scripts/cmake-build/cross_build_android_arm_inference.sh -h -
cross_build_linux_arm_inference.sh 用于ARM-Linux交叉编译。 参数 -h 可用于查询脚本支持的参数:
scripts/cmake-build/cross_build_linux_arm_inference.sh -h -
cross_build_ios_arm_inference.sh 用于IOS交叉编译。 参数 -h 可用于查询脚本支持的参数:
scripts/cmake-build/cross_build_ios_arm_inference.sh更多细节请参考 BUILD_README.md
- MegEngine 依据 贡献者公约（Contributor Covenant）来管理开源社区。请阅读 行为准则 了解更多信息。
- 每一名 MegEngine 的贡献者都需要签署贡献者许可协议（Contributor License Agreement，CLA）来明确贡献内容相关的知识产权许可。更多细节请参考 协议内容。
- 我们欢迎你通过以下方式来帮助 MegEngine 变得更好：
- 贡献代码；
- 完善文档；
- 在 MegEngine 论坛 和 Stack Overflow 回答问题；
- 在 MegEngine Model Hub 贡献新模型；
- 在 MegStudio 平台尝试新想法；
- 报告使用中的 Bugs 和 Issues；
- 审查 Pull Requests；
- 给 MegEngine 点亮小星星；
- 在你的论文和文章中引用 MegEngine；
- 向你的好友推荐 MegEngine；
- ...
我们相信我们能够搭建一个开放友善的开源社区环境，用人工智能造福人类。
- 问题: github.com/MegEngine/MegEngine/issues
- 邮箱: [email protected]
- 论坛: discuss.megengine.org.cn
- QQ: 1029741705
- OPENI: openi.org.cn/MegEngine
MegEngine 使用 Apache License, Version 2.0
Copyright (c) 2014-2020 Megvii Inc. All rights reserved.