[❓질문] 안녕하세요 프로젝트 GPU 관련 질문입니다.

[흙흙(211.244)]

https://github.com/Rem105-210/DiffFashion

GitHub - Rem105-210/DiffFashion

글쓴이 그래픽카드 : 지포스 GTX 1660 super

실행환경 : 로컬, 코랩

이 이미지 합성 프로젝트에 관심이 생겨서 모델도 다운로드하고 깃클론해서 제 로컬에서 실행했는데 자꾸 GPU를 안쓰고 CPU로 실행해서 런타임 오류가 납니다
torch.cuda.is_available() 이 명령어를 실행해도 true 리턴되는데 문제를 모르겠습니다.

그래서 코랩으로 모든 파일을 업로드한후 실행을 해도 CPU로 실행되고 hlaf연산에서 런타임 오류가 납니다.

코랩 GPU도 할당이안되니 감을 못잡겠네요

도움부탁드립니다..

[Cytrus(minhyoo1)]

args.device부분을 체크해보고 그래도 문제면 답글달아줘 - dc App

[흙흙(211.244)]

# y = self.target_transform(target).to(self.args.device) 이 파일내에서는 이코드가 주석처리되어있는데 어떤 문제인지 잘모르겠습니다 ㅜㅜ 답변갑사합니다!

[익명(118.235)]

코드에서 모델이랑 데이터를 직접 gpu로 올려야함