GPU 런타임 사용하기

코랩에서 GPU 런타임을 사용하려면 메뉴에서 런타임 > 런타임 유형 변경을 선택하고 하드웨어 가속기를 GPU로 설정하세요(그림 3-5).

▲ 그림 3-5 코랩에서 GPU 런타임 사용

텐서플로와 케라스는 GPU를 사용할 수 있다면 자동으로 GPU에서 실행됩니다. 따라서 GPU 런타임을 선택한 후 추가적으로 해야 할 일은 없습니다.

‘하드웨어 가속기’ 옵션에 TPU 런타임도 있습니다. GPU 런타임과 달리 텐서플로와 케라스에서 TPU 런타임을 사용하려면 코드에서 약간의 설정을 해 주어야 합니다. 13장에서 이에 대해 다루겠습니다. 당분간은 GPU 런타임을 사용하여 책의 코드 예제를 실행하세요.

이제 케라스 코드를 실제 실행할 준비를 마쳤습니다. 다음 절에서 2장에서 배운 핵심 아이디어가 케라스와 텐서플로 코드로 어떻게 변환되는지 알아보겠습니다.