2. 快速安装

2.1. 基于pip安装

2.1.1. 环境准备

Python

匹配目标安装包的Python版本（例如Python 3.8或者Python 3.10）
MXMACA环境

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

2.1.2. 开始安装

可在PyTorch安装压缩包中获取Triton的wheel安装包。解压PyTorch安装压缩包，可以获得Triton和PyTorch的wheel安装包（Triton 2.1版本wheel包位于PyTorch 2.1版本的安装压缩包中；Triton 3.0版本wheel包位于PyTorch 2.4版本的安装压缩包中）。

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

操作步骤

安装方式同标准whl包。

安装mcPyTorch：
```
python -m pip install torch-\*.whl
```
安装mcTriton：
```
python -m pip install triton-\*.whl
```

2.1.3. 验证安装

操作步骤

运行前设置环境变量：

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

执行以下命令：
```
python -c "import triton"
```
没有报错信息，表明Triton包安装成功。

2.1.4. 如何卸载

卸载方式同标准whl包。

操作步骤

卸载mcTriton：
```
python -m pip uninstall triton
```
卸载mcPyTorch：
```
python -m pip uninstall torch
```