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

월간 데이콘 12 | 컴퓨터 비젼 | multi-label classification | 도커 제출

  • Prize : 100만원
  • 1,046명
  • 마감
Join Competition

 

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

Follower 0 Following 0
2021.02.22 00:21 1,715 Views

최정명님이 공유해주신 코드 돌려보는데 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

감사합니다!

로그인이 필요합니다
0 / 1000