Tensorflow, Keras의 GPU 사용률 확인하기

Tensorflow, Keras의 GPU 사용률을 확인하는 방법에 대해 다루고 있습니다. 

2022. 6. 9  최초작성

학습을 진행하다가 GPU를 사용하고 있나 싶어서  작업 관리자에서 확인해보니 사용률이  거의 대부분의 시간동안 0%더라구요. 

텐서플로우 코드가 잘못되었나 싶어 검색해도 별다른 성과는 없었습니다. 

근데 이상한 점은 GPU 메모리 사용량은 5기가를 넘고 있기때문에 GPU를 사용하고 있는건 맞는 듯 싶었습니다. 

 

그래서 nvidia-smi에서 확인해보니 GPU를 사용하고 있다고 나오네요.

 

먼가 설정의 문제 인가 싶어 찾다보니 작업관리자에서 CUDA 사용량을 보도록 바꿀 수 있네요.

아래 스크린샷처럼 항목 하나를 선택하여 cuda로 바꾸면 됩니다.  

 

하지만 여전히 사용률은 0% 근처입니다.  이 값대신 cuda로 바꾼 항목을 보게되면 현재 20% 사용중이라고 보이게 됩니다. 

 

참고 

https://stackoverflow.com/questions/58289983/low-nvidia-gpu-usage-with-keras-and-tensorflow