github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-26 15:32:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

[sc demo] add requirements to spmd README (#1941)

Browse Source
This commit is contained in:
YuliangLiu0306 2022-11-14 17:22:45 +08:00 committed by GitHub
parent d5f5e06d82
commit c7925c5d08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/tutorial/auto_parallel/README.md
View File

@ -10,6 +10,12 @@ If you wish to use customized directory for the dataset. You can set the environ
export DATA=/path/to/data export DATA=/path/to/data
``` ```
## extra requirements to use autoparallel
```bash
pip install pulp
conda install coin-or-cbc
```
## Run on 2*2 device mesh ## Run on 2*2 device mesh