기존에 1개의 GPU만 사용하던 모델을 학습과 추론 단계에서 4개의 Multi GPU로 사용하도록 변경하였다. tf.device("/device:GPU:0")으로 선택하던 방식을 tensorflow 내부의 tf.distribute.mirroredstrategy 함수를 이용하여 Multi GPU를 활성화 시켰다. https://www.tensorflow.org/guide/distributed_training?hl=ko 텐서플로로 분산 훈련하기 | TensorFlow Core Google I/O는 끝입니다!
TensorFlow 세션 확인하기 세션 보기 TensorFlow 학습 TensorFlow Core 가이드 텐서플로로 분산 훈련하기 이 페이지의 내용 개요 전략의 종류 MirroredStrategy CentralStorageStrategy MultiWorkerMirroredStrategy TPUStrategy ParameterServerStrategy 케라스와 함께 tf.distrib...
원문 링크 : Multi GPU 사용
