[Python][GPU] 사용량 확인방법

여러사람이 동시에 서버를 사용할때 GPU의 사용량을 확인해서 사용하는 방법이다. 이 방법은 nvidia-smi 의 내용을 읽어들여 간단히 표현하는 방법이다.

 

pip install gpuinfo

https://pypi.org/project/gpuinfo/

gpuinfo

from gpuinfo import GPUInfo

GPUInfo.get_users(1) 

   {'mslee21': 17240}

GPUInfo.get_info()

 ==> 위의 경우value error가 발생한다. 이는 GPUInfo.py 에서 percent값을 정수로 읽어들이려 해서 발생하는 문제로 다음과 같이 바꾼다

percent.apped(int(line.splt('%')[1][-3:])

 ==> percent.apped(line.splt('%')[1][6:13])

이후 다시해보면 정상적으로 동작함