• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    미해결

cpu로 돌릴 경우, 코드 수정

20.10.05 21:04 작성 조회수 146

1

cpu로 돌릴 경우, 이 부분을 수정하면 될까요?? 

device = torch.device("cpu")

답변 1

답변을 작성해보세요.

1

안녕하세요. 킴허클베리 님

말씀하신데로 코드를 수정해서 돌리시면 됩니다. 

다만 cpu를 사용할경우에는 시간이 매우 오래 걸립니다.