본문 바로가기
IT/AI

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

by 도디누나 2022. 9. 12.
300x250

젠장...

저런 오류가 뜨길래 해결방법 찾느라 애먹었다.

아래는 해결 방법이다.

 

load_obj = torch.load("./16L_64HD_8H_512I_128T_cc12m_cc3m_3E.pt")

아래처럼 ,map_location='cpu' 를 추가하였더니 해결되었다. 

load_obj = torch.load("./16L_64HD_8H_512I_128T_cc12m_cc3m_3E.pt",map_location='cpu')

 

저처럼 코랩 돌리다 런타임 에러 뜨신분께 도움이 되길 바라며...

 

300x250