wookayin/gpustat

어떤 프로젝트인가요?

핵심 기능

• 01GPU 이름, 온도, 사용률, 메모리 사용량을 장치마다 한 줄로 표시합니다.
• 02각 GPU에서 실행 중인 프로세스와 소유자, PID, 메모리 사용량을 보여줍니다.
• 03NVIDIA(NVML), AMD(ROCm SMI), Huawei Ascend 910B 백엔드를 지원합니다.
• 04원하는 간격으로 출력을 갱신하는 watch 모드를 제공합니다.
• 05스크립트에서 사용할 수 있는 JSON 출력과 Python API를 제공합니다.

활용 사례

• 학습 작업을 시작하기 전에 공유 워크스테이션이나 서버에서 사용 가능한 GPU를 확인합니다.
• 오래 실행되는 모델 학습 중 GPU 메모리와 사용률을 모니터링합니다.
• 모니터링 스크립트에서 GPU 메트릭을 JSON으로 수집합니다.
• SSH로 접속한 환경에서 GPU 상태를 빠르고 읽기 쉬운 형식으로 확인합니다.

빠른 시작

1. 1

   PyPI에서 gpustat을 설치합니다

   pip install gpustat
2. 2

   NVIDIA GPU를 사용한다면 NVIDIA 공식 NVML 바인딩(pynvml이 아님)을 설치합니다

   pip install nvidia-ml-py
3. 3

   gpustat을 실행해 GPU 상태를 표시합니다

   gpustat

사용 전 알아둘 점

• !NVIDIA GPU를 사용하려면 NVIDIA 드라이버가 필요합니다(gpustat 1.0 이상에서는 450.00 이상).
• !nvidia-ml-py 대신 pynvml 패키지를 사용하면 작동하지 않습니다.

포함된 컬렉션

토픽

자주 묻는 질문

gpustat은(는) 무엇인가요?

NVIDIA, AMD, Huawei Ascend GPU의 사용률, 메모리, 온도, 프로세스 상태를 간결하고 읽기 쉬운 형식으로 보여주는 CLI 도구입니다. gpustat은 GPU 상태를 조회하고 모니터링하는 가벼운 Python CLI 유틸리티입니다. GPU 이름, 온도, 사용률, 메모리 사용량, 현재 실행 중인 프로세스를 장치마다 한 줄로 출력해 nvidia-smi보다 간결하게 상태를 확인할 수 있습니다.

gpustat은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?

gpustat은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.

gpustat은(는) 아직 유지보수되고 있나요?

6개월 이내 업데이트. gpustat의 마지막 커밋은 2026년 10월 8일입니다.

gpustat은(는) 어디에 사용할 수 있나요?

학습 작업을 시작하기 전에 공유 워크스테이션이나 서버에서 사용 가능한 GPU를 확인합니다. 오래 실행되는 모델 학습 중 GPU 메모리와 사용률을 모니터링합니다. 모니터링 스크립트에서 GPU 메트릭을 JSON으로 수집합니다. SSH로 접속한 환경에서 GPU 상태를 빠르고 읽기 쉬운 형식으로 확인합니다.

gpustat는 어떻게 설치하나요?

PyPI에서 gpustat을 설치합니다: pip install gpustat NVIDIA GPU를 사용한다면 NVIDIA 공식 NVML 바인딩(pynvml이 아님)을 설치합니다: pip install nvidia-ml-py gpustat을 실행해 GPU 상태를 표시합니다: gpustat

gpustat은(는) 어디서 받을 수 있나요?

소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/wookayin/gpustat