Update MACA
parent
bd9b35f410
commit
a3c150762a
22
MACA.md
22
MACA.md
@ -372,6 +372,28 @@
|
|||||||
conda install triton=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
|
conda install triton=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## 验证安装
|
||||||
|
|
||||||
|
- **运行前设置环境变量**
|
||||||
|
```bash
|
||||||
|
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
|
||||||
|
export PYTORCH_USE_FLASHATTN=1
|
||||||
|
```
|
||||||
|
|
||||||
|
- **如果使用PyTorch Extension功能,需要安装cu-bridge环境,并额外设置以下环境变量**
|
||||||
|
```bash
|
||||||
|
export CUDA_PATH=/opt/maca/tools/cu-bridge/
|
||||||
|
export CUCC_PATH=/opt/maca/tools/cu-bridge/
|
||||||
|
export PATH=${CUCC_PATH}/tools:${CUCC_PATH}/bin:${PATH}
|
||||||
|
```
|
||||||
|
- ** 执行以下命令会输出 ‘tensor([1., 1.], device='cuda:0')’ 即表明安装成功**
|
||||||
|
```bash
|
||||||
|
python -c "import torch; print(torch.ones(2).cuda())"
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
## 需要注意
|
## 需要注意
|
||||||
|
|
||||||
### 包依赖
|
### 包依赖
|
||||||
|
Loading…
Reference in New Issue
Block a user