주피터 노트북은 데이터 과학 및 머신 러닝 작업을 위한 인기 있는 통합 개발 환경입니다. 때때로 우리는 우리의 코드가 너무 많은 메모리를 사용하는지 확인하고 싶을 때가 있습니다. 이때 몇 가지 방법으로 주피터 노트북에서 현재 메모리 상태를 확인할 수 있습니다.

메모리 사용량 확인하는 방법

1. memory_usage 메서드를 사용하여 현재 메모리 사용량 확인하기

import os
import psutil

def get_memory_info():
    pid = os.getpid()
    py = psutil.Process(pid)
    memory_use = py.memory_info()[0] / 2. ** 30  # 메모리 사용량을 GB 단위로 변환
    print('주피터 노트북의 현재 메모리 사용량: {:.2f} GB'.format(memory_use))
    
get_memory_info()

2. memory_profiler 패키지를 사용하여 메모리 사용량 추적하기

memory_profiler 패키지를 사용하여 주피터 노트북에서 메모리 사용량을 추적할 수도 있습니다. 이를 위해서는 먼저 memory_profiler 패키지를 설치해야 합니다.

pip install memory_profiler

그리고 나서, 다음과 같이 memory_profiler를 사용하여 메모리 사용량을 추적할 수 있습니다.

%load_ext memory_profiler

# 메모리 추적 시작
%memit your_code_here

# 메모리 추적 중지
%memit -c your_code_here

마무리

주피터 노트북에서는 위와 같은 방법으로 메모리 사용량을 확인할 수 있습니다. 메모리 사용량을 모니터링하고 최적화하여 코드의 효율성을 높이는 것은 데이터 과학 및 머신러닝 작업에서 매우 중요합니다.

더 자세한 내용은 아래 링크를 참고하세요.

• psutil 공식 문서
• memory_profiler 공식 문서