mirror of https://github.com/hpcaitech/ColossalAI
7 lines
68 B
Plaintext
7 lines
68 B
Plaintext
|
torch>=1.8
|
||
|
|
torchvision>=0.9
|
||
|
|
numpy
|
||
|
|
tqdm
|
||
|
|
psutil
|
||
|
|
tensorboardX
|
||
|
|
packaging
|