问题

运行时两个GPU负载不均，没有跑满，效率较低
在这里插入图片描述
在这里插入图片描述

解决方法

  1. 检查数据输入，保证瓶颈部分在GPU，而不是GPU在等待
  • tf.data使用dataset.prefetch()，保证数据流及时进入模型
  • map时使用num_parallel_calls参数
  1. 增大batch_size
# python
Logo
EazyDevelop社区

一站式 AI 云服务平台

更多推荐

cover

5分钟搞定！MySQL/PostgreSQL 到 Elasticsearch 的实时同步

avatar EazyDevelop社区
cover

两小时，我搭了一套销售提成计算系统

avatar EazyDevelop社区
cover

零代码时代：如何利用聚合API平台快速构建你的专属AI Agent

avatar EazyDevelop社区