keras實現多GPU或指定GPU的使用

1. keras新版本中加入多GPU並行使用的函數 下面程序段便可實現一個或多個GPU加速: 注意:使用多GPU加速時,Keras版本必須是Keras2.0.9以上版本python from keras.utils.training_utils import multi_gpu_model #導入keras多GPU函數 import VGG19 #導入已經寫好的函數模型,例如VGG19
相關文章
相關標籤/搜索