CODEWITHSUNDEEP
pytorch-lightning
Rylan Schaeffer
Rylan Schaeffer

Reputation: 2705

PyTorch Lightning - Is Trainer necessary to use multiple GPUs?

If I want to take advantage of PyTorch Lightning's ability to train using multiple GPUs, do I have to use their Trainer?

Upvotes: 0

Views: 317

Answers (1)

Jirka
Jirka

Reputation: 1146

if you want to use all the Lightning features (even multi-GPU) such as loggers, metrics tracking, and checkpointing, then you would need to use Trainer. On the other hand, if you are fine with some limited functionality you can check out the recent LightningLite.

Upvotes: 1

Related Questions