wookayin/gpustat

해결하는 문제

gpustat은 NVIDIA GPU 상태를 모니터링하기 위해 nvidia-smi보다 더 간결하고 읽기 쉬운 대안을 제공합니다. 온도, 사용률, 메모리 사용량과 같은 GPU 지표를 단순화하여 보여주므로 어떤 프로세스와 사용자가 GPU 리소스를 소비하고 있는지 쉽게 추적할 수 있습니다.

작동 원리

NVIDIA의 공식 Python 바인딩인 nvidia-ml-py (NVML)를 래핑한 도구입니다. 명령줄 도구로 실행하거나 Python 라이브러리로 가져와 프로그래밍 방식으로 GPU 통계를 쿼리할 수 있습니다.

대상 사용자

NVIDIA GPU를 사용하는 개발자 및 연구원으로서, 하드웨어 사용률과 프로세스 소유권을 실시간으로 빠르고 명확하게 모니터링해야 하는 분들에게 적합합니다.

주요 특징

• 포괄적인 지표: GPU 인덱스, 이름, 온도, 사용률 및 메모리 사용량을 표시합니다.
• 프로세스 추적: 프로세스 소유자의 사용자 이름, 프로세스 이름 및 PID를 표시할 수 있습니다.
• 유연한 출력: 컬러 터미널 출력, 자동화를 위한 JSON 형식, 주기적인 업데이트를 위한 감시 모드를 지원합니다.
• Python API: 다른 Python 프로젝트에 통합하여 코드를 통해 GPU 통계를 쿼리할 수 있습니다.