Update MACA

repos 2024-11-06 10:36:01 +08:00
parent a3c150762a
commit 2778d22d1c

@ -382,13 +382,13 @@
export PYTORCH_USE_FLASHATTN=1
```
- **如果使用PyTorch Extension功能需要安装cu-bridge环境并额外设置以下环境变量**
- **如果使用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') 即表明安装成功**
- ** 执行以下命令,会输出 "tensor([1., 1.], device='cuda:0')" 即表明安装成功**
```bash
python -c "import torch; print(torch.ones(2).cuda())"
```