nvitop : GPU 사용 현황 확인 도구

대화형 NVIDIA 장치 및 프로세스 모니터링 도구

 

 

nvitop은 NVIDIA GPU 상태 및 프로세스를 실시간으로 모니터링하는 인터랙티브 툴입니다. NVIDIA 그래픽 드라이버 설치 시 기본 제공되는 nvidia-smi보다 풍부한 정보를 제공하며, 사용자는 GPU 장치 상태, 프로세스 필터링, 리소스 사용량, 환경 변수 등을 모니터링할 수 있습니다. 또한, CUDA 장치 선택 도구(nvisel)와 머신러닝 프레임워크(TensorFlow, PyTorch)를 위한 콜백 함수도 지원합니다.

 

주요 기능은 다음과 같습니다. 

• 실시간 리소스 모니터링
• 프로세스 필터링 및 트리 뷰 제공
• 다채로운 그래프와 유저 입력에 따른 상호작용 지원
• NVIDIA 드라이버 설치 스크립트 제공

 

실제 실행되면 다음과 같이 사용됩니다. 아래 그림은 예시이며 더 상세한 사항은 깃헙 링크 상에 설명된 내용을 참고하면 됩니다. 

 

사진 : nvitop 모니터 모드 예시

 

 

설치는 PyPI 또는 Conda를 통해 간단히 할 수 있습니다. github repo에서 수동 설치도 가능하지만 굳이 그렇게까지야... 

# PyPI에서 설치
pip install --upgrade nvitop

# conda의 경우
conda install -c conda-forge nvitop

 

 

 

사진 : nvitop의 도움말 화면 (바로가기 h)