[번외] CPU와 GPU의 학습 속도

텐서플로우는 학습시 gpu를 지원한다.

gpu를 사용하면 데이터의 병렬 처리로 인해 기존보다 빠르게 학습이 가능하다.

 

간단하게 cpu와 gpu의 속도 차이를 비교해보았다.

모델은 남성, 여성 구분 프로젝트에 사용했던 모델을 그대로 사용했다.

프로젝트에 대해 자세한 내용은 아래 링크 참조

w-storage.tistory.com/30

CNN으로 남성, 여성 구분하기

 

1. CPU

CPU만 사용했을때의 속도를 보자.

1epoch에 24초가 걸렸고 30epoch에 753.32초가 소요됐다. 약 13분정도 소요됐다.

 

2. GPU

GPU를 함께 사용했을 때의 속도를 보자.

1epoch에 2초가 걸렸고 30epoch에 63.68초가 소요됐다. 1분 조금 넘게 소요됐다.

 

3. 정리

GPU를 사용하면 CPU만 사용하는것에 비해 속도가 약 10배 이상 차이가 났다.

모델을 학습할 때 GPU를 사용하는것이 훨씬 효율적이다.

그리고 CPU에 가해지는 부하도 훨씬 적을 것이다.

CPU이용률이 100%이다...

다만 텐서플로우는 NVIDIA GPU만을 지원한다는게 다소 아쉽다.

딥러닝을 하기 위해선 NVIDIA를 계속 사용해야할 것 같다.