내 모델이 cpu에 있는지 cuda에 있는지 확인하는 방법

Tensor형태의 데이터는 .is_cuda를 붙여보면 cuda위에 있을경우 True, cpu에 있을 경우 False를 반환한다.

따라서 확인하기 쉽지만 class instance인 model은 단순하게 model.is_cuda를 할 경우 attribute error가 난다.

모델이 cuda에 있는지 확인하기 위해서는 아래 코드를 입력하면 된다.

 

next(model.parameters()).is_cuda

(여기서 model은 본인의 모델을 저장한 변수 이름으로 바꿔야한다.)