개요

케라스에서 Multi GPU를 사용하는 코드를 정리해둔다.

코드

패키지를 아래와 같이 로드한다.

from keras.utils.training_utils import multi_gpu_model

model = multi_gpu_model(model, gpus=4) 를 추가해준다.

model = cnn_model()
model = multi_gpu_model(model, gpus=4)
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(train_x, train_y, validation_data=(test_x, test_y), nb_epoch=20, batch_size=32, verbose=1)

참고자료