[이거 어떡하지] Tensor for ' ' is on CPU, Tensor for argument #1 is on CPU, but expected them to be on GPU

증상:

GPU환경에서 pytorch로 학습하는데 모델 선언하고 학습하는 때에 제목과 같은 에러가 발생합니다.

아마 주로 RNN 계열을 학습시킬 때 자주 실수하실 부분이라 생각됩니다.

 

아래와 같은 문제가 발생했을 때 어떻게 하면 될까요?

CPU가 어떻고 GPU가 어떻고..

이 에러가 무슨 말이냐면

1. model 입력은(tensor) CPU에 있는데
2. model은 GPU에 있어!

아마 RNN 계열을 학습시킬 때 hidden state 변수를 선언하실 때 실수하는 경우가 많아서(제가) 이 에러를 종종 보곤 합니다.

해결 방법

입력 tensor를 .to(device)나 .cuda()로 GPU에 올려주시면 됩니다.