ColossalAI/requirements/requirements.txt

7 lines
67 B
Plaintext

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