GPU 사용하기

GPU를 사용하면 병렬연산이 가능해서 빠르다.
CPU는 명령을 순차적으로 처리하는 것과 달리 GPU는 동시에 처리할 수 있다는 뜻이다.
연산량이 많은 딥러닝을 다룰 때는 GPU를 쓰는 것이 좋다.

GPU 사용가능 확인

import tensorflow as tf
from tensorflow.python.client import device_lib

device_lib.list_local_devices()

CPU 관련 내용과 함께 GPU 내용도 출력된 것을 확인할 수 있다. 만약 CPU에 대해서만 출력된다면 GPU를 사용할 수 없는 것이다.

tf.test.is_gpu_available()

is_gpu_available()을 셀에서 실행시켰을 때 True가 반환되면 사용 가능하다.

GPU가 있음에도 사용할 수 없다고 뜬다면?

실제로 Jupyter Notebook을 사용하고 있는데
CUDA, cuDNN을 다 설치했음에도 CPU만 떠서 왜 그런가 했더니
Jupyter notebook을 다시 켜고 코드를 입력하니 사용가능 해졌었다.