mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-26 01:35:21 +00:00
c20529fe78
* [examples] update autoparallel tutorial demo * add test_ci.sh * polish * add conda yaml
12 lines
262 B
Bash
12 lines
262 B
Bash
#!/bin/bash
|
|
set -euxo pipefail
|
|
|
|
conda init bash
|
|
conda env create -f environment.yaml
|
|
conda activate auto
|
|
cd ../../..
|
|
pip uninstall colossalai
|
|
pip install -v .
|
|
cd ./examples/tutorial/auto_parallel
|
|
colossalai run --nproc_per_node 4 auto_parallel_with_resnet.py -s
|