mirror of https://github.com/hpcaitech/ColossalAI
28 lines
836 B
Markdown
28 lines
836 B
Markdown
|
# Setup
|
||
|
|
|
||
|
|
## Announcement
|
||
|
|
|
||
|
|
Our auto-parallel feature is a alpha version. It is still under development. We will keep updating it and make it more stable. If you encounter any problem, please feel free to raise an issue.
|
||
|
|
|
||
|
|
## Requirements
|
||
|
|
|
||
|
|
We need some extra dependencies to support auto-parallel. Please install them before using auto-parallel.
|
||
|
|
|
||
|
|
### Install PyTorch
|
||
|
|
|
||
|
|
We only support PyTorch 1.12 now, other versions are not tested. We will support more versions in the future.
|
||
|
|
|
||
|
|
```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
|
||
|
|
```
|
||
|
|
|
||
|
|
### Install pulp and coin-or-cbc
|
||
|
|
|
||
|
|
```bash
|
||
|
|
pip install pulp
|
||
|
|
conda install -c conda-forge coin-or-cbc
|
||
|
|
```
|