Update MACA

repos 2024-11-05 16:30:21 +08:00
parent 36306b03d9
commit 071e581c1e

34
MACA.md

@ -309,35 +309,65 @@
``` ```
```bash ```bash
使用 conda 安装 或者使用 conda 安装
conda install torch==<version> -c https://repos.metax-tech.com/r/maca-conda/ conda install torch=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
``` ```
- **2. 安装 mcFlashattn** - **2. 安装 mcFlashattn**
```bash ```bash
使用 pip 安装
pip install flash_attn==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com pip install flash_attn==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com
``` ```
```bash
或者使用 conda 安装
conda install flash_attn=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
```
- **3.安装 mcTorchvision(可选)** - **3.安装 mcTorchvision(可选)**
```bash ```bash
使用 pip 安装
pip install torchvision==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com pip install torchvision==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com
``` ```
```bash
或者使用 conda 安装
conda install torchvision=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
```
- **4.安装mcTorchaudio(可选)** - **4.安装mcTorchaudio(可选)**
```bash ```bash
使用 pip 安装
pip install torchaudio==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com pip install torchaudio==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com
``` ```
```bash
或者使用 conda 安装
conda install torchaudio=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
```
- **5.安装 mcApex(可选)** - **5.安装 mcApex(可选)**
```bash ```bash
使用 pip 安装
pip install apex==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com pip install apex==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com
``` ```
```bash
或者使用 conda 安装
conda install apex=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
```
- **6.安装 mcTriton(可选)** - **6.安装 mcTriton(可选)**
```bash ```bash
使用 pip 安装
pip install triton==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com pip install triton==<version> i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com
``` ```
```bash
或者使用 conda 安装
conda install triton=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
```
### 需要注意 ### 需要注意
#### 包依赖 #### 包依赖