分类：多显卡

2020
10-07

keras多显卡训练方式

使用keras进行训练，默认使用单显卡，即使设置了os.environ['CUDA_VISIBLE_DEVICES']为两张显卡，也只是占满了显存，再设置tf.GPUOptions(allow_growth=True)之后可以清楚看到，只占用了第一张显卡，第二张显卡完全没用。要使用多张显卡，需要按如下步骤：（1）importmulti_gpu_model函数：fromkeras.utilsimportmulti_gpu_model（2）在定义好model之后，使用multi_gpu_model设置模型由几张显卡训练，如下：model=Model(...)#... 继续阅读 >