If you're having difficulties installing DeepSpeed, check the DeepSpeed CUDA installation guide.