这边在 Colab 的 GPU 上跑了一下没有问题,建议参考 InvalidArgumentError: assertion failed: [0] [Op:Assert] name: EagerVariableNameReuse · Issue #38518 · tensorflow/tensorflow · GitHub ,以及检查你的 GPU 是不是被其他程序占用了。
这边在 Colab 的 GPU 上跑了一下没有问题,建议参考 InvalidArgumentError: assertion failed: [0] [Op:Assert] name: EagerVariableNameReuse · Issue #38518 · tensorflow/tensorflow · GitHub ,以及检查你的 GPU 是不是被其他程序占用了。