좋아요는
1분 내에 한 번만 클릭 할 수 있습니다.
제 2회 컴퓨터 비전 학습 경진대회
월간 데이콘 12 | 컴퓨터 비젼 | multi-label classification | 도커 제출
- Prize : 100만원
- 1,046명
- 마감
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
닉네임
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
List
이전 글
질문있습니다.
Competition -
제 2회 컴퓨터 비전 학습 경진대회
Likes 8
Views 1,507
Comments 1
9 months ago
현재 글
gpu 메모리 한계 질문입니다,
Competition -
제 2회 컴퓨터 비전 학습 경진대회
Likes 8
Views 1,715
Comments 7
9 months ago
다음 글
[필독] 데이터 재배포 공지
Competition -
제 2회 컴퓨터 비전 학습 경진대회
Likes 5
Views 2,132
Comments 8
10 months ago
혹시 Docker 환경을 사용중이신가요 ? Docker 컨테이너 구성간 GPU 메모리 공유를 낮게하신게 아닐까요 ?