上传者: 38640443
|
上传时间: 2021-09-10 18:18:02
|
文件大小: 114KB
|
文件类型: PDF
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 m