python 코드 메모리 사용량, 시간 계산 (memory profiler)

패키지 설치는 아래와 같이 Linux cmd창에 입력

pip install memory_profiler

분석하고 싶은 python 코드 안에 들어가서 실행 부분을 감싸고 위에 @profile 추가.

from memory_profiler import profile 

@profile
def run():

	###함수나 클래스 등 실행할 코드 혹은 분석하고 싶은 부분을 감쌈###


if __name__ == '__main__':

    run()

 

그 후 다시 Linux cmd창에 아래와 같이 입력. (process.py 부분에 위와 같이 데코레이터를 넣은 파이썬 파일 넣기)

python -m memory_profiler process.py

from codarchive

위와 같이 line by line별로 메모리 사용량과 증가량이 나타난다.

메모리 사용량과 수행 시간 그래프로 그리기

 mprof run process.py

위를 통해 .dat파일을 생성해준다

 mprof plot -o image.png

그 후 plot 옵션을 통해 그래프를 그리는데 이때 plot에서 따로 지정해주지 않으면 가장 최근에 생성된 .dat파일을 통해 그래프를 그림.

from codarchive

그러면 위와 같이 메모리 사용량과 시간이 나온다.