pip安装

您可根据以下步骤安装MegEngine

安装说明

您可以通过包管理器 pip 安装 MegEngine:

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

再在 python 中导入 megengine 验证安装成功:

import megengine as mge import megengine as mge

目前 MegEngine 安装包集成了使用 GPU 运行代码所需的 CUDA 10.1 环境,不区分 CPU 版本和 GPU 版本。如果您想运行 GPU 程序,请保证机器本身配有 NVIDIA 显卡,并且 驱动 版本高于 418.x 。

对于大部分用户,通过包管理器安装打包完毕的 MegEngine 足够应对所有使用需求了,但是如果需要使用最近更新还未发布的特性,则可能需要从源码编译安装。另外如果对 模型部署 有需求或者希望参与到 MegEngine 的核心开发工作中,也需要了解从源码进行安装。详细内容请参考 源码编译安装

注意

MegEngine 目前支持 Linux 、Windows、Mac 环境下安装,支持Python3.5 及以上的版本(不支持 Python2 )。

对于 Windows 10 用户,建议通过安装 WSL(Windows Subsystem for Linux) 进行体验。

京ICP备19000496号-11

用户协议| 隐私政策