Tensorflow CPU/GPU 목록 확인하기

텐서플로우로 코드를 짜다보면, 가끔 with.device로 특정 CPU 또는 GPU를 지정할 때가 있습니다. 본인이 쓰고 있는 하드웨어에 있는 디바이스 목록을 확인하는 방법이 있는데요.

---

아래와 같이 코드를 입력하시면, 현재 디바이스에서 사용가능한 cpu/gpu 목록을 볼 수 있습니다.

1 2

from tensorflow.python.client import device_lib print(device_lob.list_loca_devices())

cs

저 같은 경우, 아래와 같이 프린트 내역이 나타나네요.

---

[name: "/device:CPU:0"
device_type: "CPU"
# ... 중략
. name: "/device:GPU:0"
# ...
]