mirror of https://github.com/hpcaitech/ColossalAI
29 lines
822 B
Markdown
29 lines
822 B
Markdown
|
# 安装
|
|||
|
|
|
|||
|
|
## 声明
|
|||
|
|
|
|||
|
|
我们的自动并行功能处于alpha版本，仍在快速的开发迭代中。我们会在兼容性和稳定性上做持续地改进。如果您遇到任何问题，欢迎随时提issue给我们。
|
|||
|
|
|
|||
|
|
|
|||
|
|
## 要求
|
|||
|
|
|
|||
|
|
我们需要一些额外的依赖性来支持自动并行功能。 请在使用自动平行之前安装它们。
|
|||
|
|
|
|||
|
|
### 安装PyTorch
|
|||
|
|
|
|||
|
|
我们仅支持Pytorch 1.12，现在未测试其他版本。 将来我们将支持更多版本。
|
|||
|
|
|
|||
|
|
```bash
|
|||
|
|
#conda
|
|||
|
|
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
|
|||
|
|
#pip
|
|||
|
|
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
### 安装pulp和coin-or-cbc
|
|||
|
|
|
|||
|
|
```bash
|
|||
|
|
pip install pulp
|
|||
|
|
conda install -c conda-forge coin-or-cbc
|
|||
|
|
```
|