Cannot pin 'torch.cuda.LongTensor' only dense CPU tensors can be pinned

Shxck69 October 3, 2023, 7:05pm 1

Have an error when i try to train the model using a gpu with pytorch

1 Like

Topic		Replies	Views
Using Trainer with LayoutXLM for classification Beginners	1	1163	May 3, 2023
T5 Model For Summarization On GPU Beginners	0	160	February 1, 2024
RuntimeError: Input, output and indices must be on the current device Beginners	0	406	March 8, 2023
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:3! (when checking argument for argument index in method wrapper_CUDA__index_select) Models	0	252	December 25, 2023
Trainer.train throws RuntimeError: Expected all tensors to be on the same device Beginners	5	2467	May 17, 2023