Setting specific device for Trainer

ndvb April 10, 2023, 12:34pm 10

Same issue. Why when loading a model using .from_pretrained we can specify device_map, but we can’t specify device_map when defining a Trainer?

Topic		Replies	Views
How to set gpu device for hugging trainer? 🤗Transformers	1	1097	September 16, 2024
How to restrict training to one GPU if multiple are available, co 🤗Transformers	4	14419	November 1, 2023
How to set the training device to cuda:1 ? By default, TrainerArgument seems to move model to cuda:0 Beginners	3	342	September 25, 2024
How to restrict Trainer to use certain GPUs? Beginners	2	535	February 25, 2024
Limit GPU cores for training 🤗Transformers	4	1545	September 14, 2023