2. 快速安装
2.1. 基于pip安装
2.1.1. 环境准备
Python
匹配目标安装包的Python版本
MXMACA环境
安装Driver软件包以及MXMACA SDK软件包
PyTorch环境
安装PyTorch软件包
2.1.2. 开始安装
mcApex安装包以 apex-${APEX_VERSION}+metax${MACA_VERSION}-${PYTHON_VERSION}-${PYTHON_VERSION}-${arch_info}.whl 格式命名。
操作步骤
安装方式同标准whl包。
安装mcApex:
python -m pip install apex-*.whl
2.1.3. 验证安装
操作步骤
运行前设置环境变量:
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/
执行以下命令,没有报错即表明安装成功:
python -c "import apex"
2.1.4. 如何卸载
卸载方式同标准whl包。
操作步骤
卸载mcApex:
python -m pip uninstall apex
2.2. 使用Docker运行
mcApex依赖mcPyTorch运行,在MXMACA-PyTorch镜像中提供了预装的mcApex。MXMACA-PyTorch镜像以离线形式发布,用户可在随本文档发布的软件包中找到相关压缩包。
操作步骤
获取MXMACA容器镜像,执行以下命令,完成容器镜像的加载:
docker load < ./${container_name}.container.xz
在Docker容器中使用板卡,使用全部曦云系列GPU(以C500为例):
docker run -it --device=/dev/mxcd --device=/dev/dri --group-add video ${image_name} /bin/bash