텐서플로우(Tensorflow) gpu 사용 가능 여부 확인하기

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

[name: "/device:CPU:0"

device_type: "CPU"

memory_limit: 268435456

locality { }

incarnation: 10360034786306285900

xla_global_id: -1,

name: "/device:GPU:0"

device_type: "GPU"

memory_limit: 7787773952

locality { bus_id: 1 links { }

}

]