Files
ColossalAI/requirements/requirements.txt
Jiarui Fang 54229cd33e [log] better logging display with rich (#426)
* better logger using rich

* remove deepspeed in zero requirements
2022-03-16 09:51:15 +08:00

10 lines
84 B
Plaintext

torch>=1.8
torchvision>=0.9
numpy
tqdm
psutil
tensorboard
packaging
pre-commit
rich