Hugging Face Forums
Setting specific device for Trainer
Beginners
rodio
July 10, 2023, 4:19am
15
The same issue still exist, literally dealing with this right now
show post in topic
Related topics
Topic
Replies
Views
Activity
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
14421
November 1, 2023
How to set the training device to cuda:1 ? By default, TrainerArgument seems to move model to cuda:0
Beginners
3
343
September 25, 2024
How to restrict Trainer to use certain GPUs?
Beginners
2
535
February 25, 2024
Limit GPU cores for training
🤗Transformers
4
1546
September 14, 2023