Google Colab CPU와 GPU의 속도 차이 확인

Google Colab으로 딥러닝 학습 시 GPU의 성능이 더 좋다고 합니다.

그렇다면 CPU와 비교하여 성능이 어느 정도 차이 나는지 궁금하여 예제를 찾아 확인해보았습니다.

https://colab.research.google.com/notebooks/gpu.ipynb#scrollTo=Y04m-jvKRDsJ&uniqifier=1

위의 링크를 참고하였으며, 임의의 이미지 위에 컨볼루션 신경망 계층을 구성하고 결과 ops를 CPU 또는 GPU에 수동으로 배치하여 실행 속도를 비교하는 예제입니다.

 

아래는 실행 결과입니다.

GPU가 CPU 대비 87배 빠르다고 합니다. 처리에 오랜 시간이 걸리는 딥러닝 모델을 학습할 때는 하드웨어 가속기의 GPU를 켜주는 것이 좋겠습니다. 

또한 실행할 때마다 작업 수행시간이 다르게 출력되었는데요, 이는 하드웨어 자원을 다른 Colab 사용자들과 함께 사용하기 때문이라고 합니다.

감사합니다.