文件名称:FlexFlow:分布式深度学习框架,支持灵活的并行化策略
文件大小:623KB
文件格式:ZIP
更新时间:2024-04-22 01:24:45
C++
弹性流 FlexFlow是一个深度学习框架,可通过自动搜索有效的并行化策略来加速分布式DNN培训。 FlexFlow提供了TensorFlow Keras和PyTorch的替代产品。 在FlexFlow中运行现有的Keras和PyTorch程序只需要对该程序进行。 安装FlexFlow 要从源代码安装FlexFlow,请阅读。 如果您想快速尝试FlexFlow,我们还提供了预先安装了所有依赖项的预构建docker。 您还可以使用conda安装FlexFlow Python包(即将推出)。 TensorFlow Keras支持 用户只需更改以下导入标题行,即可使用FlexFlow加快现有TensorFlow Keras模型的训练过程。 from flexflow . keras . models import Model , Sequential from flexflow . kera