mirror of https://github.com/hpcaitech/ColossalAI
4 lines
253 B
Bash
4 lines
253 B
Bash
|
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
|
||
|
|
pip install colossalai==0.1.10+torch1.12cu11.3 -f https://release.colossalai.org
|
||
|
|
pip install titans
|
||
|
|
colossalai check -i
|