Selecting one gpu from cli #8063
Answered
by
tshu-w
alexyalunin
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
Hi, I have 4 gpus on my machine. I want to select the third one, that has index 2. How do I pass an argument from CLI? |
Beta Was this translation helpful? Give feedback.
Answered by
tshu-w
Jun 21, 2021
Replies: 1 comment
-
Maybe you should search first |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
alexyalunin
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
python 4_pretrain_encoder.py --gpus 2, --max_epochs 5
Maybe you should search first
#7461