반응형


nvidia-smi 초마다 자동 갱신하기

 

머신러닝 개발등을 하다보면, 

GPU 메모리 상태와 사용량등을 보여주는 nvidia-smi 명령어를 통해 지속적으로 학습이 잘 진행되는지 확인할 일이 많다.

 

 

하지만 nvidia-smi는 불행히도 자동 갱신이 되지 않고, 

nvidia-smi라는 커맨드를 입력한 당시의 GPU 상태만 한번 프린트 해주는 커맨드이다.

그러다보니, 초단위로 GPU 상태가 어떻게 변하는지 확인하기 위해서는  nvidia-smi를 지속적으로 입력해줘야 한다.

이러한 일을 간단하게 해주는 ubuntu의 watch 기능이 존재한다.

 

## 1초마다 갱신
watch -n 1 nvidia-smi

## 5초마다 갱신
watch -n 5 nvidia-smi

 

위 명령어를 실행하게 되면, 아래와 같이 초단위로 갱신되는 nvidia-smi 결과를 확인할 수 있다.

 

 

반응형
블로그 이미지

Hyunsoo Luke HA

석사를 마치고 현재는 Upstage에서 전문연구요원으로 활동중인 AI 개발자의 삽질 일지입니다! 이해한 내용을 정리하는 용도로 만들었으니, 틀린 내용이 있으면 자유롭게 의견 남겨주세요!

,