Ubuntu 에서 GPU 사용량 확인 및 메모리 삭제

 안녕하십니까 코딩오페라의 Master.M입니다. 오늘은  Ubuntu 환경에서 GPU 사용량을 확인 및 메모리 삭제하는 방법에 대해 알아보도록 하겠습니다.(Ubuntu 20.04 버전)

GPU 사용량 확인

 우선  Ubuntu 환경에서 'CTRL + ALT + T'를 눌러 터미네이터 창을 켜줍니다. 그런다음 아래와 같은 코드를 입력해 줍니다.  

 

nvidia-smi -l 1

 

 

입력을 해주면 위와 같은 창이 나옵니다. 여기서 빨강색 창을 보시면 해당 컴퓨터의 GPU용량과 현재 사용량을 보실수 있습니다. 이 창은 매초 업데이트가 되기 때문에 실시간으로 확인이 가능합니다. 

 

GPU 메모리 삭제

 위와 같이 Ubuntu 환경에서 터미네이터 창을 켜줍니다. 그런다음 아래와 같은 코드를 입력해 줍니다.  

 

# 현재 사용중인 프로그램 확인
ps aux | grep python

 

 

입력을 해주면 위와 같은 창이 나옵니다. 여기에는 지금 실행중인 프로그램 목록들이 나오는데 그 중에서 jupyter-notebook 프로그램을 GPU 메모리에서 지우도록 하겠습니다. 보시다 싶이 해당 프로그램의 번호는 '4372' 입니다. 

 

# 우분투에서 실행중인 프로그램 삭제
# sudo kill -9 '해당 프로그램 번호'
sudo kill -9 4372

 

다시 터미널 창으로가서 해당 프로그램을 위의 코드를 이용해 삭제해주면 됩니다. 삭제후 GPU 메모리 사용량을 확인해보면 아래와 같이 여유가 생긴 모습을 알 수 있습니다. 

 

 

이로써 'Ubuntu 에서 GPU 사용량 확인 및 메모리 삭제' 글을 마무리 하도록 하겠습니다. 도움이 되셨거나 궁금한 사항은 밑에 댓글을 이용해 주시기 바랍니다. 감사합니다. 저는 코딩오페라의 Master.M이었습니다.