2. 快速安装

2.1. 基于pip安装

2.1.1. 环境准备

  • Python

    匹配目标安装包的Python版本

  • MXMACA环境

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

2.1.2. 开始安装

可在PyTorch安装压缩包中获取Triton的wheel安装包。 解压PyTorch安装压缩包,可以获得Triton和PyTorch的wheel安装包。

PyTorch安装压缩包以 maca-pytorch${pytorch_version}-py${python_version}-${release_version}-${arch_info}.tar.xz 格式命名。 Triton和PyTorch的wheel安装包分别以 triton-*.whltorch-*.whl 格式命名。

操作步骤

安装方式同标准whl包。

  1. 安装mcPyTorch:

    python -m pip install torch-\*.whl
    
  2. 安装mcTriton:

    python -m pip install triton-\*.whl
    

2.1.3. 验证安装

操作步骤

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

    export MACA_PATH=/opt/maca/
    export LD_LIBRARY_PATH=${MACA_PATH}/lib:${MACA_PATH}/mxgpu_llvm/lib: ${MACA_PATH}/ompi/lib:${LD_LIBRARY_PATH}
    export MACA_CLANG_PATH=${MACA_PATH}/mxgpu_llvm/bin
    
  2. 执行以下命令:

    python -c "import triton"
    

    没有报错信息,表明Triton包安装成功。

2.1.4. 如何卸载

卸载方式同标准whl包。

操作步骤

  1. 卸载mcTriton:

    python -m pip uninstall triton
    
  2. 卸载mcPyTorch:

    python -m pip uninstall torch