Hello, thank you for your work. I encountered a problem while running the code. I want to train using a single GPU, but I'm having trouble selecting which card to use. I have tried some methods, but it always uses the first card. Where should I make the modification?
Hello, thank you for your work. I encountered a problem while running the code. I want to train using a single GPU, but I'm having trouble selecting which card to use. I have tried some methods, but it always uses the first card. Where should I make the modification?