repos/index
1
0
Code Issues Actions Packages Projects Wiki

Update MACA

repos 2024-11-06 10:36:01 +08:00
parent a3c150762a
commit 2778d22d1c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
MACA.md

@ -382,13 +382,13 @@
export PYTORCH_USE_FLASHATTN=1 export PYTORCH_USE_FLASHATTN=1
``` ```
- **如果使用PyTorch Extension功能，需要安装cu-bridge环境，并额外设置以下环境变量** - **如果使用PyTorch Extension功能,需要安装cu-bridge环境,并额外设置以下环境变量**
```bash ```bash
export CUDA_PATH=/opt/maca/tools/cu-bridge/ export CUDA_PATH=/opt/maca/tools/cu-bridge/
export CUCC_PATH=/opt/maca/tools/cu-bridge/ export CUCC_PATH=/opt/maca/tools/cu-bridge/
export PATH=${CUCC_PATH}/tools:${CUCC_PATH}/bin:${PATH} export PATH=${CUCC_PATH}/tools:${CUCC_PATH}/bin:${PATH}
``` ```
- ** 执行以下命令会输出 ‘tensor([1., 1.], device='cuda:0')’ 即表明安装成功** - ** 执行以下命令,会输出 "tensor([1., 1.], device='cuda:0')" 即表明安装成功**
```bash ```bash
python -c "import torch; print(torch.ones(2).cuda())" python -c "import torch; print(torch.ones(2).cuda())"
``` ```