jupyter notebook 개별 메모리 사용량 확인하는 방법

광고 한 번씩 눌러주세요! 블로그 운영에 큰 힘이 됩니다 :)

memory_profiler를 설치하면 됨.

!pip install memory_profiler

 

magic 함수로 확인하는 방법

%load_ext memory_profiler
%memit

 

패키지로 확인하는 방법

 

from memory_profiler import memory_usage
mem_usage = memory_usage(-1, interval=1, timeout=1)
print(mem_usage)

https://data-newbie.tistory.com/312?category=772752

Jupyter Notebook에서 Notebook 이름 가져오기

위의 글에서 주피터 노트북 이름을 가져오고 중간중간 logging하면 해당 주피터 메모리 체크할 수 있을것 같다.

 

 

def current_memory_check() :
    from memory_profiler import memory_usage
    mem_usage = memory_usage(-1, interval=1, timeout=1)[0]
    usage = f"{mem_usage:.3f} MiB"
    return usage
    

def current_notebook_name() :
    import ipyparams
    notebook_name = ipyparams.notebook_name
    return notebook_name

 

 

 

https://pypi.org/project/memory-profiler/

memory-profiler