2. 快速安装

2.1. 基于pip安装

2.1.1. 环境准备

  • Python

    匹配目标安装包的Python版本(例如,Python 3.8或者Python 3.10)

  • MXMACA环境

    安装Driver软件包以及MXMACA SDK软件包

  • PyTorch环境

    安装PyTorch软件包

2.1.2. 开始安装

mcApex安装包以 apex-${APEX_VERSION}+metax${MACA_VERSION}-${PYTHON_VERSION}-${PYTHON_VERSION}-${arch_info}.whl 格式命名。 例如,mcApex + Python 3.8版本的安装包名称为: apex-0.1+metax2.33.0.3-cp38-cp38-linux_x86_64.whl

操作步骤

安装方式同标准whl包。

  1. 安装mcApex:

    python -m pip install apex-*.whl
    

2.1.3. 验证安装

操作步骤

  1. 运行前设置环境变量:

    export MACA_PATH=/opt/maca/
    export LD_LIBRARY_PATH=${MACA_PATH}/lib:${MACA_PATH}/mxgpu_llvm/lib:${LD_LIBRARY_PATH}
    export MACA_CLANG_PATH=${MACA_PATH}/mxgpu_llvm/bin/
    
  2. 执行以下命令,没有报错即表明安装成功:

    python -c "import apex"
    

2.1.4. 如何卸载

卸载方式同标准whl包。

操作步骤

  1. 卸载mcApex:

    python -m pip uninstall apex
    

2.2. 使用Docker运行

mcApex依赖mcPyTorch运行,在MXMACA-PyTorch镜像中提供了预装的mcApex。MXMACA-PyTorch镜像以离线形式发布,用户可在随本文档发布的软件包中找到相关压缩包。

本文档中以 mxc500-torch2.6-py310-mc2.33.0.3-ubuntu24.04-amd64.container.xz 为例,用户应根据实际收到的软件包版本对版本字段进行相应替换。

操作步骤

  1. 获取MXMACA容器镜像,执行以下命令,完成容器镜像的加载:

    docker load < ./mxc500-torch2.6-py310-mc2.33.0.3-ubuntu24.04-amd64.container.xz
    
  2. 在Docker容器中使用板卡,使用全部曦云系列GPU(以C500为例):

    docker run -it --device=/dev/mxcd --device=/dev/dri --group-add video mxc500-torch2.6-py310:mc2.33.0.3-ubuntu24.04-amd64 /bin/bash