如何安装 MegEngine¶
注解
MegEngine 目前支持在以下环境安装 Python 包:
操作系统: Linux-64 位 / Windows-64 位 / MacOS-10.14 及其以上
Python 版本:3.6 到 3.9
其中 MacOS 只支持 Intel x86 CPU; 此外,MegEngine 也支持在很多其它平台上进行推理运算。
通过包管理器安装¶
通过 pip
包管理器安装 MegEngine 的命令如下:
python3 -m pip install --upgrade pip # 将 pip 更新到最新版本
python3 -m pip install megengine -f https://megengine.org.cn/whl/mge.html
注解
对于 conda
用户, 可以选择通过在环境中先安装 pip
, 再按照上述方式进行 MegEngine 的安装;
警告
MegEngine 包中集成了 CUDA 环境,但用户需确保环境中已经正确地安装好 GPU 设备 相关驱动。
通过源码编译安装¶
如果包管理器安装的方式无法满足你的需求,例如:
我使用的 GPU 设备非 Nvidia 厂商的,比如用的是 AMD 等厂商的 GPU;
我使用的 Nvidia GPU 设备比较新或者比较旧,不在当前的设备支持列表中;
我希望更改一些其它的编译配置选项,启用一些默认关闭的特性。
则可以尝试自行通过源码编译安装。相关细节请参考 scripts/cmake-build/BUILD_README.md .
CUDA 编译支持现状¶
MegEngine CMake CUDA 编译的现状如下:
CUDA 11.1 及以上编译能适配市面上所有的 Ampere 卡,适配 sm80+sm86
CUDA 11.0 编译能适配 A100, 但不能适配 30 系卡,仅适配 sm80
CUDA 10 不适配 Ampere 架构(官方发布的预编译包是基于 CUDA 10.1 的)
注解
用户可以使用 cmake -DMGE_CUDA_GENCODE="-gencode arch=compute80, code=sm80"
自由指定。
警告
用户在编译前需要确定有 GPU 设备,以及确定环境中所使用的 CUDA 版本。
参见
用户可在 Compute Capability 页面找到自己的 GPU 设备对应的计算兼容性版本。