OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 Commits
28 Branches
25 MB
3465 Download
 
 
 
 
 
 
Tree: e3de2011c0
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'e3de2011c0'
${ noResults }
Megvii Engine Team e3de2011c0 build(mge): clang compile and link error 5 years ago
.github ci(workflow): add gpu ci workflow 5 years ago
ci ci(workflow): add gpu ci workflow 5 years ago
cmake MegEngine: Initial commit of MegEngine. 5 years ago
dnn build(mge): clang compile and link error 5 years ago
python_module feat(mgb/gopt): add nchw44 optpass 5 years ago
sdk feat(mgb/gopt): add nchw44 optpass 5 years ago
src build(mge): clang compile and link error 5 years ago
test feat(mgb/core): add comp node for cambricon 5 years ago
third_party MegEngine: Initial commit of MegEngine. 5 years ago
tools/param_defs MegEngine: Initial commit of MegEngine. 5 years ago
.gitattributes MegEngine: Initial commit of MegEngine. 5 years ago
.gitignore MegEngine: Initial commit of MegEngine. 5 years ago
.gitmodules MegEngine: Initial commit of MegEngine. 5 years ago
ACKNOWLEDGMENTS MegEngine: Initial commit of MegEngine. 5 years ago
CMakeLists.txt fix(mge/build): remove MEGDNN flags from command line 5 years ago
CODE_OF_CONDUCT.md MegEngine: Initial commit of MegEngine. 5 years ago
CONTRIBUTOR_LICENSE_AGREEMENT.md MegEngine: Initial commit of MegEngine. 5 years ago
LICENSE MegEngine: Initial commit of MegEngine. 5 years ago
README.md doc(mge): add openi msg 5 years ago
README_CN.md doc(mge): add openi msg 5 years ago
logo.png MegEngine: Initial commit of MegEngine. 5 years ago

README.md

MegEngine

MegEngine Logo

English | 中文

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

Installation

NOTE: MegEngine now only supports Linux platform with Python 3.5 or higher. On Windows 10 you could try WSL(Windows Subsystem for Linux) to use Linux within Windows.

Binaries

Commands to install from binaries via pip wheels are as follows:

pip3 install megengine -f https://megengine.org.cn/whl/mge.html

Build from Source

Prerequisites

Most of the dependencies of MegEngine are located in third_party directory, and you do
not need to install these by yourself. you can prepare these repositories by executing:

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

But some dependencies should be manually installed:

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

Build

MegEngine prefers Out-Of-Source flavor, and compile in a mostly-static way.
Here are the instructions:

  1. Make a directory for the build.

    mkdir -p build
cd build

  2. Generate build configurations by CMake.

    For CUDA build:

    cmake .. -DMGE_WITH_TEST=ON

    For CPU only build, use -DMGE_WITH_CUDA=OFF:

    cmake .. -DMGE_WITH_CUDA=OFF -DMGE_WITH_TEST=ON

    For deployment with C++ only, use -DMGE_INFERENCE_ONLY=ON, and turn off test with -DMGE_WITH_TEST=OFF:

    cmake .. -DMGE_INFERENCE_ONLY=ON -DMGE_WITH_TEST=OFF

    Use -DCMAKE_INSTALL_PREFIX=YOUR_PATH to specify the install path.

  3. Start to build.

    make -j$(nproc)

  4. [optional] Install the library if compiled for deployment at step 2.

    make install

Here are some other useful options for the build.

  • MGE_ARCH specifies which arch MegEngine are building for. (default AUTO)
  • MGE_WITH_DISTRIBUTED if multiple machine distributed support is enabled. (default ON)
  • MGE_WITH_PYTHON_MODULE if build python module. (default ON)
  • MGE_BLAS chooses MKL or OpenBLAS as BLAS library for MegEngine. (default MKL)
  • MGE_CUDA_GENCODE supplies the -gencode option for nvcc. (default not supply)
  • MGE_DISABLE_FLOAT16 if disable float16 support. (default OFF)
  • MGE_ENABLE_EXCEPTIONS if enable exception support in C++. (default ON)
  • MGE_ENABLE_LOGGING if enable logging in MegEngine. (default AUTO)

More options can be found by:

cd build
cmake -LAH .. 2>/dev/null| grep -B 1 'MGE_' | less

How to Contribute

We believe we can build an open and friendly community and power humanity with AI.

How to contact us

Resources

License

MegEngine is Licensed under the Apache License, Version 2.0

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

Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境，不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序，请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉，欢迎访问 MegStudio 平台

C++ Cuda Python C SVG other