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.
2271 Commits
28 Branches
25 MB
3208 Download
 
 
 
 
 
 
Tree: 1add4517ad
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '1add4517ad'
${ noResults }
Megvii Engine Team 1add4517ad test(trace): test subtensor on unknown shape 3 years ago
.github feat(ci): download ninja 3 years ago
ci fix(ci): add MGE_WITH_CUSTOM_OP in github ci 3 years ago
cmake feat(cmake): format all cmake file 3 years ago
dnn feat(dnn,mgb,imperative): add diag opr implement 3 years ago
imperative test(trace): test subtensor on unknown shape 3 years ago
lite feat(lite): add global layout transform python interface for lite 3 years ago
scripts fix(imperative): fix use collections error from python3.10 3 years ago
sdk fix(sdk/load_and_run): remove docs of dump_with_testcase_with_mge.py 3 years ago
src feat(dnn,mgb,imperative): add diag opr implement 3 years ago
test feat(cmake): format all cmake file 3 years ago
third_party feat(mge): add cpp_redis and tacopie submodule 3 years ago
toolchains feat(cmake): format all cmake file 3 years ago
tools feat(cmake): format all cmake file 3 years ago
.clang-format style(dnn): add bypass of clang-format for dnn foreach_opr macro 3 years ago
.gdbinit feat(dispatch): add gdb debug utils 3 years ago
.gitattributes test(imperative/python): fix testcase for magicmind runtime module 3 years ago
.gitignore chroe(.gitignore): add more ignore rules for git 4 years ago
.gitmodules feat(mge): add cpp_redis and tacopie submodule 3 years ago
ACKNOWLEDGMENTS feat(lite): add header licence 3 years ago
CMakeLists.txt fix(cmake): enable custom op when building develop to avoid the pytest fail 3 years ago
CODE_OF_CONDUCT.md fix typos 3 years ago
CONTRIBUTOR_LICENSE_AGREEMENT.md MegEngine: Initial commit of MegEngine. 5 years ago
LICENSE chore(mge): update copyright years 4 years ago
README.md doc(mge): add bibtex msg 3 years ago
README_CN.md doc(mge):update readme 3 years ago
commitlint.config.js ci(copybara): add config files and fix format script 3 years ago
logo.png MegEngine: Initial commit of MegEngine. 5 years 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 --upgrade pip
python3 -m pip install megengine -f https://megengine.org.cn/whl/mge.html

Building from Source

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

Citation

If you use MegEngine in your publication,please cite it by using the following BibTeX entry.

@Misc{MegEngine,
  institution = {megvii},
  title =  {MegEngine:A fast, scalable and easy-to-use deep learning framework},
  howpublished = {\url{https://github.com/MegEngine/MegEngine}},
  year = {2020}
}

Copyright (c) 2014-2021 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