How to run train script on cuda:1 instead of cuda:0 #235
Unanswered
SarthakJaingit
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi,
Recently, I was looking to train an effecient det model; however, in a shared server only cuda:1 is open. I was wondering what option is there that I can choose to train on cuda:1 instead of cuda:0 and not get a CUDA out of memory error.
Beta Was this translation helpful? Give feedback.
All reactions