Skip to content

Does gluonts Trainer supports multi-GPU training? #2348

Answered by lostella
la-cruche asked this question in Q&A
Discussion options

You must be logged in to vote

This is not supported in MXNet-based models (the training loop we use does not support that currently). For PyTorch models, we rely on Lightning for training, so there it may be possible (but I haven't done it, so this needs to be validated).

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by la-cruche
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants