DeepSpeed
Use torch.cuda.device_count()
#126
Merged

Loading