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

最新推荐文章于 2024-09-03 23:05:30 发布

shuangyumelody

最新推荐文章于 2024-09-03 23:05:30 发布

阅读量2.3w

点赞数 6

分类专栏： keras 文章标签：多GPU加速指定GPU加速 keras

本文链接：https://blog.csdn.net/qq_36427732/article/details/79017835

版权

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