본문 바로가기

프로그래밍/AI:ML:DL

GPU 사용 모니터 nvidia-smi-gui

반응형

ubuntu 에서 2020년/2022년에 실행확인한 내용임.

AI 훈련할 때 gpu 를 사용하고 있는건지 모니터해 보고 싶을 때가 있다.

nvidia-smi 명령으로 많이 확인하는데, 텍스트 화면을 읽으려니 내겐 가독성이 너무 나빴다. gui 그래픽 윈도우를 띄워서 확인해 보고 싶은 욕구가 드는데, 이걸 만들어 놓은 사람이 있다.

github  nvidia-smi-gui 로 검색하면 레포지토리( https://github.com/imkzh/nvidia-smi-gui )가 있고, 레포 clone 해서 간단한 몇가지 디펜던시만 설치해 주고 실행하면 다음과 같이 예쁜 윈도우가 뜬다.

간단하게 사용할 수 있으니 한번씩 사용해 보길 바란다.

나는 sh 파일을 하나 만들고, /usr/local/bin/ 에 링크해 놓아서, 간단히 명령만 입력하면 모니터 윈도우가 실행되도록 해 놓았다.


실행하기 위한 준비작업

  • sudo apt-get install python3-pyqt5
  • conda create -n nvidiasmigui  : 전용 파이썬 가상환경을 anaconda 로 만듦.
  • conda activate nvidiasmigui  : 해당 가상환경으로 들어감.
  • (nvidiasmigui) conda install -c conda-forge pyqt : pyqt 패키지를 conda 로 설치. github 에서는 pyqt4 가 필요하다고 하지만, 소스 보면, pyqt5 를 임포트하여 사용하고 있음. pyqt 패키지가 설치되면 pyqt5로 설치됨 (2022년 4월 현재)
  • (nvidiasmigui) python nvidia-smi-gui.py

 

728x90