RuntimeError: Cuda out of memory

Pear_Mh·2022년 8월 3일
0

딥러닝을 할 때 가장 많이 보이는 에러입니다. 오죽하면 다른 오류보다 해당 오류를 보면 반가울 정도..

메모리 사용량 초과 이슈입니다.
보통 nvidia-smi를 통해 구동되는 process를 확인하는데, 구동되는 process는 없지만 사용량이 가득할 때는 아래 방식을 이용하면 됩니다.
방법 1

import gc
import torch
gc.collect()
torch.cuda.empty_cache()

방법 2 - 서버의 상태 확인 후 사용 할 것

# 주의 터미널도 꺼짐
# pkill -u [userid]
$ pkill -u 0

# 혹은
$ pkill -u (root)
profile
Beyond the new era.

0개의 댓글