[Keras] Tensorflow GPU 사용여부 확인하기

기존의 코드

import tensorflow as tf

tf.test.is_gpu_available()

(결과)

- 위와 같이 'True'가 나오면 동작하는 것으로 확인할 수 있음

---

 

해당 코드를 실행하면 Warning(2022.01.28 기준)이 발생한다. 이는 위의 코드가 더이상 사용되지 않을 수 있으니 다른 코드를 사용하라고 나타났다.

---

새로 사용할 코드

import tensorflow as if

tf.config.list_physical_devices('GPU')

(결과)

사용가능한 GPU가 리스팅된다. 이는 tf를 통해 접근이 가능한 것이므로, 작동이 되는 것으로 볼 수 있다.