ColossalAI/examples/tutorial/hybrid_parallel/requirements.txt

3 lines
36 B
Plaintext
Raw Normal View History

colossalai >= 0.1.12
torch >= 1.8.1