[ 에러 ] RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'

 

에러 명

RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'

 

 

yolact 모델을 학습 하는데 이러한 에러가 떴습니다.

gpu 와 cpu 를 통한 학습 관련 부분에서 에러가 뜬거 같은데요.

해결방법을 알아보도록 하겠습니다.

 

해결방법

 

 

torch\utils\data\sampler.py 파일에서 115번째 줄 에 있는

generator = torch.Generator() 라고 되어있는 부분을

generator =torch.Generator(device='cuda') 로 바꾸어 주면 됩니다. 

 

 

그리고, 저는 yolact 모델 학습 train.py 파일에서

data_loader 부분의 마지막에 

data_loader = data.DataLoader(... generator=torch.Generator(device='cuda')) 를 추가해주어 해결했습니다.