How to use all the available GPUs #12755
-
Hi all, I migrated my code to the latest version of lightnining and tried using UserWarning: The flag What is the recommended way to use, by default, all GPUs (when present)? |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 1 reply
-
Hi @mnslarcher! To use all of available devices, use: Trainer(..., devices="auto") The doc page is available here: https://pytorch-lightning.readthedocs.io/en/latest/common/trainer.html#devices |
Beta Was this translation helpful? Give feedback.
-
For anyone seeing this discussion, have a look at #12756 (comment). |
Beta Was this translation helpful? Give feedback.
For anyone seeing this discussion, have a look at #12756 (comment).