[AI] Jupyter에서 GPU 사용 여부 확인하기

GPU를 사용하면 tensorflow에서 모델을 효과적으로 학습할 수 있습니다.

간단한 수식을 통해 각 텐서와 연산이 어떠한 장치에 할당되었는지를 출력해서 현재 내 컴퓨터에서 GPU를 사용하고 있는지, CPU를 사용하고 있는지를 확인할 수 있습니다.

 

import tensorflow as tf

tf.debugging.set_log_device_placement(True)

a = tf.constant([
	[1,1],
    [2,2]
])

b = tf.constant([
	[5,5],
    [7,7]
])

c = tf.matmul(a,b)
print(c)

tf.debugging.set_log_device_placement(False)

이러한 GPU 관련 내용이 나온다면 GPU가 사용되고 있는 것입니다.

 

다른 방법으로는,

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

 

이처럼 CPU와 GPU 모두 나온다면 정상적으로 사용가능합니다.