ColossalAI/requirements/requirements.txt

10 lines
80 B
Plaintext

torch>=1.8
numpy
tqdm
psutil
packaging
pre-commit
rich
click
fabric
contexttimer