목록으로

gpu 메모리 한계 질문입니다,

#제 2회 컴퓨터 비전 학습 경진대회

최정명님이 공유해주신 코드 돌려보는데 vram이 bach = 10만해도 꽉 차는데 이유가 월까요?

2070super에서도 47로 돌아가는데 3080인데도 batch가 10이 한계네요 ㅠㅠ

ubuntu 18.04에 pytorch/pytorch:1.7.1~..runtime dockerfile 사용중입니다.

  • 핑크퐁 핑크퐁 2021.02.22 10:25
    혹시 Docker 환경을 사용중이신가요 ? Docker 컨테이너 구성간 GPU 메모리 공유를 낮게하신게 아닐까요 ?
    • 닉네임 닉네임 2021.02.22 11:07
      제가 docker를 처음 써보는데요
      ubuntu 18.04에서 docker pytorch/pytorch로 사용할 때 nvidia-smi를 해보면 vram이 10기가로 꽉 차고
      윈도우에서 도커 없이 아나콘다 가상환경으로 했을 때도 batch가 10 되니깐 vram이 10기가로 꽉 차더라구요
      공유 코드는 batch_size 말고는 수정 안했습니다.
  • 닉네임 닉네임 2021.02.22 11:32
    삭제된 댓글입니다
  • 닉네임 닉네임 2021.02.22 11:32
    해결했습니다! 
    • 최정명 최정명 2021.02.22 12:07
      혹시 코드상에 문제가 있었나요 ?
      어떻게 해결 하셨나요 ?
      
    • 닉네임 닉네임 2021.02.22 16:41
      제가 B7 쓰고 싶어서 그걸로 해놔서 안되는 거였어요.
      B0로 하니깐 잘 되었습니다.
    • 최정명 최정명 2021.02.22 16:55
      아 그렇군요 ! 감사합니다.
      
      B0, B3, B7을 돌려봤는데 점점 성능이 좋아짐을 했습니다.
      
      오래 걸리는게 문제긴 하지만 ㅠㅠ 전처리를 통해 성능을 향상 시킨다음 깊은 모델로 학습하면 좋은 결과가 있을것 같네요.
    • 닉네임 닉네임 2021.02.22 17:06
      감사합니다!
로그인이 필요합니다로그인이 필요합니다
목록으로