딥러닝코드를 돌릴 때 GPU는 굉장히 중요합니다. GPU 사용량을 확인하기 위해서 nvidia-smi를 실행해서 확인하죠. 우분투 환경에서 nvidia-smi 라고 터미널에 입력하면 nvidia-smi가 실행되고 GPU 사용량을 알 수 있습니다. 그런데 nvidia-smi 를 입력할 때의 사용량만 보여주지 실시간으로 어떻게 변하는지 보여주질 않습니다. 이번 글에서는 GPU 사용량을 실시간으로 사용할 수 있도록 하는 방법에 대해 알아보겠습니다. 그럼 시작해보겠습니다.
nvidia-smi 실시간으로 GPU 사용량 확인
아래와 같은 공식을 터미널에 입력하면 됩니다.
watch -n 1 nvidia-smi
위와 같이 터미널에 입력하면 1초 간격으로 nvidia-smi상태를 보여줍니다. 만약에 2초간격으로 보고 싶다면 아래와 같이 1을 2로 바꾸기만 하면 됩니다.
watch -n 2 nvidia-smi