Files
ColossalAI/requirements/requirements.txt
2021-10-28 18:21:23 +02:00

7 lines
68 B
Plaintext

torch>=1.8
torchvision>=0.9
numpy
tqdm
psutil
tensorboardX
packaging