How to set the training device to cuda:1 ? By default, TrainerArgument seems to move model to cuda:0
I have 2 gpus. And I want to train separate models on both of them. Which is why i need to move one model to cuda:1
2024-09-24T19:43:14Z
I figured this one out. All I had to do was setCUDA_VISIBLE_DEVICES = 1and setdevice_map = "auto"when loading the model.
2024-09-24T22:06:49Z
model.to(“cuda:1”) will also work.
2024-09-25T10:07:22Z
This topic was automatically closed 12 hours after the last reply. New replies are no longer allowed.