본문 바로가기
Etc

[2021.08.05] How to solve memory errors(Out of memory) on jupyter notebook?

by injekim97 2021. 8. 5.
반응형

[2021.08.05] How to solve memory errors(Out of memory) on jupyter notebook?

 

 

 

주피터 노트북을 사용하다가 , 메모리 에러(Out of memory)가 발생할 수 있다.

해결 방법 :  아래와 같은 코드를 주피터 노트북에서 실행해 주면 된다.

 

pip install torch

-> 우선 위의 torch 라이브러리를 pip를 통해 설치해 준다.

 

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

-> 위의 모듈을 설치 한 후에,  위의 코드를 주피터 노트북에 적어서, 실행 시켜주면 메모리 정리가 된다. 

 

주피터 메모리 해결 방법 Out of memory.ipynb
0.00MB

반응형

댓글