tensorflow - 如何使用多个 GPU 运行深度排序？

Question

我正在关注本教程。我有两个 nvidia 1080 Ti 显卡，但我无法输入代码。

我在第 8 行有错误我还有一台带有一个 GPU 的笔记本电脑，该代码可以正常工作

我稍微更改了代码。

tf.config.experimental.set_memory_growth(physical_devices[0:1], True)

但仍然是同样的错误

Answer 1

感谢社区我找到了答案

而不是使用 tf.config.experimental.set_memory_growth(physical_devices[0:1], True)

我使用 tf.config.experimental.set_memory_growth(physical_devices[0], True) tf.config.experimental.set_memory_growth(physical_devices[1], True)

我的代码在我的 gpu 上运行良好并且现在没有出现值错误