keras实现多GPU或指定GPU的使用

最新推荐文章于 2024-07-29 09:08:28 发布

原创

最新推荐文章于 2024-07-29 09:08:28 发布 · 2.3w 阅读

30 ·

CC 4.0 BY-SA版权

文章标签：

#多GPU加速 #指定GPU加速 #keras

1. keras新版本中加入多GPU并行使用的函数

下面程序段即可实现一个或多个GPU加速：
注意：使用多GPU加速时，Keras版本必须是Keras2.0.9以上版本

from keras.utils.training_utils import multi_gpu_model   #导入keras多GPU函数
import VGG19     #导入已经写好的函数模型，例如VGG19

if G <= 1:
    print("[INFO] training with 1 GPU...")
    model = VGG19（）

# otherwise, we are compiling using multiple GPUs
else:
    print("[INFO] training with {} GPUs...".format(G))
    # we'll store a copy of the model on *every* GPU and then combine
    # th