分类：指定GPU

2020
10-08

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

1.keras新版本中加入多GPU并行使用的函数下面程序段即可实现一个或多个GPU加速：注意：使用多GPU加速时，Keras版本必须是Keras2.0.9以上版本fromkeras.utils.training_utilsimportmulti_gpu_model#导入keras多GPU函数importVGG19#导入已经写好的函数模型，例如VGG19ifG<=1:print("[INFO]trainingwith1GPU...")model=VGG19（）#otherwise,wearecompilingusingmultipleGPUselse:print("[INFO]train... 继续阅读 >