[GPU] Tensorflow GPU 사용가능 여부 확인

 

Tensorflow 설치 시 GPU를 사용하기 위해 CUDA, cuDNN 등을 추가 설치하여 환경을 구성하는데

모두 진행했음에도 불구하고 Tensorflow에서 GPU가 사용되지 않는 경우가 있다.

 

아래 가이드는 Tensorflow 환경 구성 후 GPU를 사용할 수 있는 상태인지 체크하는 방법이다.

 

 

Python 실행 후 아래 명령어 실행

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

 

결과

노란색 부분처럼 device에 GPU로 보이는 항목이 보이면 Tensorflow에서 사용 가능하다.