Deep Learning 개발 중 현재 GPU를 점유하는 pid 및 GPU 사용량 확인하는 방법

 

머신러닝 프로젝트나 개발을 하면서 GPU 작업을 시작하면, Iteration이 올라가지 않거나 GPU 작업만 멈춘 경우를 볼 수 있다.

또는 GPU 작업간 현재 사용량 정보를 확인할 필요가 있을 경우 htop 등과 다르게 GPU 유관 정보가 필요할때 유용하다.

 

Ubuntu의 기본적인 nvidia-smi 설치 경로

Ubuntu에 nvidia driver를 설치했다면, 기본적으로 link 되어 있다. /etc/bin 쪽에 연결되어 있다. 

Cloud 기반의 GPU가 pre-setup된 VM을 이용한다면, 일반적으로 걱정할 필요 없이 즉시 실행 가능하다.

 

실행 / bash 또는 윈도우 cmd에서 연속실행

nvidia-smi_1.jpg

 

이렇게 실행하고 결과를 확인 가능.

 

윈도우에서 연속 실행

아래와 같이 anyname.bat로 bat 파일을 만들고, 실행

 

:LOOP
nvidia-smi
ping 127.0.0.1 -n 3 >NUL
goto :LOOP

 

 

만약, path에 없다면, 적당히 시스템 환경 변수에 path를 걸어주자.

 

bash에서 연속 실행

간결하게 가능하다

 

while :; do clear; nvidia-smi; sleep 3; done

 

 

참고링크 

How do I run nvidia-smi on Windows? - Stack Overflow

Where can I find nvidia-smi.exe utility - CUDA / CUDA Programming and Performance - NVIDIA Developer Forums

Useful nvidia-smi commands - Learn CUDA Programming (packtpub.com)

 

 

No. Subject Author Date Views
Notice SQL강좌: 챗GPT와 함께 배우는 SQL Server 무료 강좌 목차와 소개 (2023년 9월 업데이트) 코난(김대우) 2023.08.18 24028
Notice Python 무료 강좌 - 기초, 중급, 머신러닝(2023년 6월 업데이트) 코난(김대우) 2021.01.01 12459
14 Python 머신러닝 강좌 - 5. Pandas DataFrame에 질의(Query) 코난(김대우) 2021.01.07 257
13 Python 머신러닝 강좌 - 4. Pandas DataFrame file 코난(김대우) 2021.01.07 216
12 Python 머신러닝 강좌 - 3. Pandas 소개 file 코난(김대우) 2021.01.07 302
11 Python 머신러닝 강좌 - 2. Anaconda와 Conda 구성 file 코난(김대우) 2021.01.06 309
10 Jupyter notebook 단축키 등의 cheat sheet file 코난(김대우) 2021.01.06 326
9 Python 머신러닝 강좌 - 1. 주피터 노트북(Jupyter Notebook) 구성 file 코난(김대우) 2021.01.06 884
8 2021년 머신러닝과 인공지능(AI) 트렌드 - MLaaS (서비스로의 머신러닝) file 코난(김대우) 2020.12.29 385
7 머신러닝을 시작하는 개발자를 위한 - (4) 2021년의 ML Trend - MLaaS와 MLOps(Machine Learning + DevOps) 코난(김대우) 2020.12.28 389
6 머신러닝을 시작하는 개발자를 위한 - (3) 우리 개발자가 머신러닝을 해야 하는 이유는? 코난(김대우) 2020.12.28 233
5 머신러닝을 시작하는 개발자를 위한 - (2) 머신러닝 서비스/프레임워크/툴킷 분류 및 전체 개발 흐름 코난(김대우) 2020.12.28 391
4 머신러닝을 시작하는 개발자를 위한 - (1) 머신러닝 용어정리, 분류부터 MLOps까지 file 코난(김대우) 2020.12.28 645
» Deep Learning 개발 중 현재 GPU를 점유하는 pid 및 GPU 사용량 확인하는 방법 file 코난(김대우) 2020.12.26 284
2 GPU 머신에서 Machine Learning을 수행할때 Deep Learning 툴킷 & 패키지들이 GPU를 이용하는지 체크하는 python 코드 코난(김대우) 2020.12.21 218
1 머신러닝 새로운 글 코난(김대우) 2020.12.18 257





XE Login