我有一台带有 2 个 GPU 的虚拟机。我正在使用 tf.data API 进行模型训练。我想用一个 GPU 处理数据，另一个用于训练。
基本上，为了处理预处理操作，我在 with tf.device(device_name): 之前引入了这行代码。
这同样不适用于训练，其中同一行代码似乎无法正常工作。
有人知道解决我的问题的最佳方法吗？
提前致谢！
我有一台带有 2 个 GPU 的虚拟机。我正在使用 tf.data API 进行模型训练。我想用一个 GPU 处理数据，另一个用于训练。
基本上，为了处理预处理操作，我在 with tf.device(device_name): 之前引入了这行代码。
这同样不适用于训练，其中同一行代码似乎无法正常工作。
有人知道解决我的问题的最佳方法吗？
提前致谢！