위 코드를 통해 TPU가 제대로 연결됐는지 확인할 수 있습니다. TPU의 네트워크 주소 경로와 TPU 코어 개수가 제대로 나타난다면 성공입니다. TPU를 확인했다면 TPU 사용을 위한 ‘strategy’를 만듭니다. ‘strategy’를 사용하면 텐서플로나 tf.keras로 만들어진 모델을 TPU 학습이 가능하도록 바꿀 수 있습니다.
Note ≣ | TPU 사용 시 주의 사항
TPU 연결을 시도할 때 다음과 같은 주의 사항을 볼 수 있습니다.
▲ 그림 7-16 TPU 사용 시 주의 사항
요약하면, TPU를 올바르게 사용하기 위해서는 문서(Documentation)를 한번 숙지하는 것이 좋고, TPU는 구글 클라우드 스토리지(GCS)의 데이터를 읽기 때문에 인터넷 옵션이 자동으로 켜지게 된다는 뜻입니다. 만약 인터넷 사용이 금지된 대회의 경우, 사용하지 못할 수 있기 때문에 꼭 확인하라는 의미입니다.