How to specify only a single non-zero-indexed GPU? #19855
Unanswered
NielsRogge
asked this question in
code help: NLP / ASR / TTS
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
I'm on a shared device and would like to only leverage GPU with index = 1.
How to do this with Lightning?
I tried passing devices="1", devices=[1] but it's still leveraging the first 2 GPUs, as confirmed by this error:
My code is this notebook.
Beta Was this translation helpful? Give feedback.
All reactions