安装 MegEngine¶
注解
MegEngine 目前支持在以下环境安装 Python 包:
操作系统: Linux-64bit/Windows-64bit/MacOS-10.14 及其以上
Python 版本:3.5 到 3.8
其中 MacOS 只支持 Intel x86 CPU; 此外,MegEngine 也支持在很多其它平台上进行推理运算。
通过包管理器安装¶
通过 pip
包管理器安装 MegEngine 的命令如下:
python3 -m pip install megengine -f https://megengine.org.cn/whl/mge.html
注解
对于 conda
用户, 可以选择通过在环境中先安装 pip
,
再按照上述方式进行 MegEngine 的安装。
通过源码编译安装¶
如果包管理器安装的方式无法满足你的需求,则可以尝试自行通过源码编译安装。
环境依赖¶
大多数编译 MegEngine 的依赖位于 third_party 目录,可以通过以下命令自动安装:
./third_party/prepare.sh
./third_party/install-mkl.sh
上述过程中需要从国外获取软件源,建议使用比较通畅的网络环境。
一些依赖需要手动安装:
开始编译¶
scripts/cmake-build/host_build.sh 用于本地编译。
参数
-h
可用于查询脚本支持的参数:scripts/cmake-build/host_build.sh -h
scripts/cmake-build/cross_build_android_arm_inference.sh 用于 ARM-安卓 交叉编译。
参数
-h
可用于查询脚本支持的参数:scripts/cmake-build/cross_build_android_arm_inference.sh -h
scripts/cmake-build/cross_build_linux_arm_inference.sh 用于 ARM-Linux 交叉编译。
参数
-h
可用于查询脚本支持的参数:scripts/cmake-build/cross_build_linux_arm_inference.sh -h
scripts/cmake-build/cross_build_ios_arm_inference.sh 用于 iOS 交叉编译。
参数
-h
可用于查询脚本支持的参数:scripts/cmake-build/cross_build_ios_arm_inference.sh -h
更多细节请参考 scripts/cmake-build/BUILD_README.md .