首页 > 解决方案 > 不能在 tf.data 管道中使用 .map_on_gpu()

问题描述

我正在使用 tf.data API 来训练 AI 模型,我发现 TensorFlow 有 .map_on_gpu() 允许直接在 GPU 上映射数据集。

有谁知道如何使用它?TensorFlow v2.4.1 似乎不允许公开使用。

提前致谢!

标签: pythontensorflow

解决方案


推荐阅读