ColossalAI/examples/tutorial/hybrid_parallel/requirements.txt

3 lines
36 B
Plaintext

colossalai >= 0.1.12
torch >= 1.8.1