1 Star 2 Fork 1

AX0424 / MegEngine

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

MegEngine

MegEngine Logo

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

但是有一些依赖需要手动安装:

  • 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 遵循“源外构建”(Out-of-Source Build)原则,并且使用静态编译方式。编译的具体流程如下:

  1. 创建用于编译的目录:

    mkdir -p build
    cd build
  2. 使用 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 指定具体安装目录。

  3. 开始编译:

    make -j$(nproc)
  4. [可选] 如果需要用于部署,可以安装 MegEngine 的 C++ 库:

    make install

以下是其它常用编译选项:

  • MGE_ARCH 指定编译的目标平台(默认自动检测当前平台)
  • MGE_WITH_DISTRIBUTED 是否开启多机分布式支持(默认开启)
  • MGE_WITH_PYTHON_MODULE 是否编译生成 Python 模块(默认开启)
  • MGE_BLAS 选择 BLAS 的后端实现,可以是 MKLOpenBLAS (默认 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 依据 贡献者公约(Contributor Covenant)来管理开源社区。请阅读 行为准则 了解更多信息。
  • 每一名 MegEngine 的贡献者都需要签署贡献者许可协议(Contributor License Agreement,CLA)来明确贡献内容相关的知识产权许可。更多细节请参考 协议内容
  • 我们欢迎你通过以下方式来帮助 MegEngine 变得更好:

我们相信我们能够搭建一个开放友善的开源社区环境,用人工智能造福人类。

联系我们

资源

开源许可

MegEngine 使用 Apache License, Version 2.0

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

空文件

简介

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架 展开 收起
C++
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
C++
1
https://gitee.com/arthurxu424/MegEngine.git
git@gitee.com:arthurxu424/MegEngine.git
arthurxu424
MegEngine
MegEngine
master

搜索帮助

14c37bed 8189591 565d56ea 8189591