天元(MegEngine)是旷视自主研发的开源深度学习框架,于2020年3月正式在 OpenI 启智社区开源,天元能够帮助开发者高效的完成深度学习算法的设计、训练、部署,有效提升AI研发工作效率。 https://megengine.org.cn/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Megvii Engine Team 2d18074a70 fix(mgb): fix spell error 2 months ago
.github ci(imperative): run imperative python test 9 months ago
ci ci(imperative): run imperative python test 9 months ago
cmake feat(cmake/arm/cuda): support cmake arm with cuda, 2 months ago
dnn fix(mgb): fix spell error 2 months ago
imperative fix(mgb): fix spell error 2 months ago
scripts fix(dnn): fix cudnn crash when finalize called after cudnn dtor 2 months ago
sdk fix(mgb): fix spell error 2 months ago
src fix(mgb): fix spell error 2 months ago
test feat(mgb/opr): add fast profile and combined Execution strategy 2 months ago
third_party chore(third_party): update upstream of cutlass 2 months ago
toolchains feat(cmake/arm/cuda): support cmake arm with cuda, 2 months ago
tools fix(mgb): fix spell error 2 months ago
.gitattributes feat(dnn/cuda): add deconv int8 and fix cutlass conv wrapper base on modify cutlass 2.4 2 months ago
.gitignore chroe(.gitignore): add more ignore rules for git 9 months ago
.gitmodules chore(third_party): add range-v3 as third_party dependency 5 months ago
ACKNOWLEDGMENTS chore(mge/license): remove multipledispatch from ACKNOWLEDGMENTS 5 months ago
CMakeLists.txt fix(dnn): fix cudnn crash when finalize called after cudnn dtor 2 months ago
CODE_OF_CONDUCT.md MegEngine: Initial commit of MegEngine. 1 year ago
CONTRIBUTOR_LICENSE_AGREEMENT.md MegEngine: Initial commit of MegEngine. 1 year ago
LICENSE chore(mge): update copyright years 4 months ago
README.md Update README and fix file permissions. 3 months ago
README_CN.md Update README and fix file permissions. 3 months ago
logo.png MegEngine: Initial commit of MegEngine. 1 year ago

README.md

MegEngine

English | 中文

MegEngine is a fast, scalable and easy-to-use deep learning framework, with auto-differentiation.


Installation

NOTE: MegEngine now supports Python installation on Linux-64bit/Windows-64bit/MacOS(CPU-Only)-10.14+ platforms with Python from 3.5 to 3.8. On Windows 10 you can either install the Linux distribution through Windows Subsystem for Linux (WSL) or install the Windows distribution directly. Many other platforms are supported for inference.

Binaries

To install the pre-built binaries via pip wheels:

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

Building from Source

Prerequisites

Most of the dependencies of MegEngine are located in third_party directory, which can be prepared by executing:

./third_party/prepare.sh
./third_party/install-mkl.sh

But some dependencies need to be installed manually:

  • CUDA(>=10.1), cuDNN(>=7.6) are required when building MegEngine with CUDA support.
  • TensorRT(>=5.1.5) is required when building with TensorRT support.
  • LLVM/Clang(>=6.0) is required when building with Halide JIT support.
  • Python(>=3.5) and numpy are required to build Python modules.

Build

MegEngine uses CMake as the build tool.
We provide the following scripts to facilitate building.

  • host_build.sh builds MegEngine that runs on the same host machine (i.e., no cross compiling).
    The following command displays the usage:
    scripts/cmake-build/host_build.sh -h
    
  • cross_build_android_arm_inference.sh builds MegEngine for DNN inference on Android-ARM platforms.
    The following command displays the usage:
    scripts/cmake-build/cross_build_android_arm_inference.sh -h
    
  • cross_build_linux_arm_inference.sh builds MegEngine for DNN inference on Linux-ARM platforms.
    The following command displays the usage:
    scripts/cmake-build/cross_build_linux_arm_inference.sh -h
    
  • cross_build_ios_arm_inference.sh builds MegEngine for DNN inference on iOS (iPhone/iPad) platforms.
    The following command displays the usage:
    scripts/cmake-build/cross_build_ios_arm_inference.sh
    

Please refer to BUILD_README.md for more details.

How to Contribute

We strive to build an open and friendly community. We aim to power humanity with AI.

How to Contact Us

Resources

License

MegEngine is Licensed under the Apache License, Version 2.0

Copyright (c) 2014-2021 Megvii Inc. All rights reserved.