ColossalAI/examples/tutorial/auto_parallel/test_ci.sh

7 lines
168 B
Bash

#!/bin/bash
set -euxo pipefail
pip install -r requirements.txt
conda install -c conda-forge coin-or-cbc
colossalai run --nproc_per_node 4 auto_parallel_with_resnet.py