Tensorflow가 GPU를 지원하는지 알아보자

Tensorflow를 설치해서 사용하다보면 현재 설치된 환경이 GPU를 제대로 지원하는지 아닌지 궁금할 때가 있습니다. 요즘은 설치가 많이 편해지기는 했지만, 리눅스 버전, GPU driver 버전, TF 버전, Cuda 버전 등 모두 맞추기가 쉽지 않았으니까요. TF를 설치를 해도 나도 모르게 CPU 버전으로 돌아가는 경우가 왕왕 있습니다.

(GPU Tensroflow는 Docker를 이용하는 것을 추천합니다. Docker를 이용해서 편리하게 GPU Tensorflow를 사용하자)

현재 환경이 GPU를 지원하는지 어떻게 알 수 있을까요? 구글링을 해보면 TF 1.x 시절부터해서 여러가지 방법이 나옵니다. 현재 TF 2.x 기준으로 가장 간략한 방법은 다음 코드를 실행해보는 것입니다.

import tensorflow as tf

tf.config.list_physical_devices('GPU')

만약에 GPU 지원이 활성화되어 있다면,

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

처럼 나올 겁니다.

아니라면,

[]

이렇게 빈 list가 출력됩니다.

혹시나 이 방법이 통하지 않는다면 아래 명령으로 전체 디바이스 목록을 볼 수도 있습니다.

tf.python.client.device_lib.list_local_devices()