一、实验

      采用keras的multi_gpu_model函数注释中的示例代码

多GPU真的能加速吗?

  经过实验发现,当batch_size=4的时候,2GPU速度为30s/epoch,4GPU速度为36s/epoch。速度反而下降了

      当batch_size=32时,2GPU速度为10s/epoch,4GPU速度为8s/epoch,速度1.25倍

      当batch_size=64时,2GPU速度为9s/epoch,4GPU速度为6s/epoch,速度1.5倍

二、结论

     多GPU模式存在通信开销,batch_size越大,GPU计算开销所占比例越大,通信开销所占比例越小,多GPU的速度提高越接近准线性

相关文章:

  • 2022-12-23
  • 2021-05-08
  • 2022-12-23
  • 2022-12-23
  • 2021-06-14
  • 2021-08-05
  • 2022-12-23
  • 2022-12-23
猜你喜欢
  • 2021-08-11
  • 2021-05-18
  • 2021-07-27
  • 2021-04-07
  • 2021-04-20
  • 2021-10-03
  • 2021-08-02
相关资源
相似解决方案