Hugging Face Forums

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!

🤗Transformers

lucasjin June 5, 2023, 7:25am 8

I think I know why, am I solved it.

Topic		Replies	Views	Activity
Multi-GPU finetuning of NLLB produces RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0 🤗Transformers	1	631	August 29, 2023
Can I use CUDA with Trainer.train? Beginners	3	6075	May 10, 2022
Device_map="auto" with error: Expected all tensors to be on the same device Beginners	6	3989	March 12, 2024
Pipeline on GPU Beginners	0	252	October 15, 2023
Trainer.train throws RuntimeError: Expected all tensors to be on the same device Beginners	5	2531	May 17, 2023