欢迎关注博主的公众号:happyGirl的异想世界。有更多干货还有技术讨论群哦~

  1. from keras.utils.multi_gpu_utils import multi_gpu_model
  2. os.environ['CUDA_VISIBLE_DEVICES'] = '2,3'#用序号为2和3的卡
  3. model=ourModel(image_size, depth=depth, k=k)()
  4. model = multi_gpu_model(model,2)#参数2的意思就是我们用两块卡