GPU out of memory

월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회

GPU out of memory

2022.04.18 10:19 3,037 Views

안녕하세요~ 대회 참여를 이제 시작해보려 하는데 질문사항이 있습니다

베이스라인 코드를 실행하고 있는데,,, 분명 colab에서 잘 돌아가는데 jupyter에서 하려하면 out of memory가 뜨네요 ㅠ

구글링해서 하라는 건 다 해봤는데 (empty_cache 등) 안 되는지라,,, 혹시 저와 같은 오류가 생기시는 분이 있을까요?

batch_size를 16이나 8로 낮춰도 안되고, 된다 하더라도 f1 score가 좋아지지 않는,,, ㅠ

CPU : i7-9750HF

메모리 : 64GB

GTX : 1660ti

로그인이 필요합니다

comment

0 / 1000

도플라밍고

2022.04.18 13:34

지금 로컬에서 사용하시는 GPU의 메모리가 코랩의 GPU 메모리보다 딸려서 그럴수도 있지 않을까요!?

Pobredward

2022.04.19 09:10

아 맞네요,,, 1660ti가 좋은 줄 알았는데 GPU램이 6GB입니다,,, 코랩은 12GB더라고요 ^ㅠ^

도플라밍고

2022.04.18 13:35

device = 'cpu' 로 하셔도 out of memory가 뜨나요..?.?

ㅇㅇㅇㅇ--

2022.04.21 21:24

혹시 colab에 필요한 패키지 설치하시면서 dependency 문제 발생하시지 않으셨나요??ㅜㅜ

현재 글

GPU out of memory

Competition - 월간 데이콘 Computer Vision 이상치 탐지 알고리즘 경진대회

Likes 11

Comments 4

3년 전