안녕하세요,
재재입니다.
Tensorflow 연산 GPU 동작되는지,
아니면 CPU 에서 동작되는지 쉽게 확인 할 수 있는 방법을 설명드릴게요.
(1) 간단한 코드 실행
import tensorflow as tf
tf.debugging.set_log_device_placement(True)
# generates variables
a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]])
# do operation
c = tf.matmul(a, b)
print(c)
(2) 실행결과 확인
tensorflow gpu에서 동작되는 경우 아래와 같고,
Executing op MatMul in device /job:localhost/replica:0/task:0/device:GPU:0
tensorflow cpu에서 동작되는 경우에는,
Executing op MatMul in device /job:localhost/replica:0/task:0/device:CPU:0
이렇게 로그를 받게 됩니다.
tensorflow 연산 GPU 동작 확인하기