如下在创建模型时出现的错误

model = tf.keras.Sequential()

报错内容为:

tensorflow.python.framework.errors_impl.InvalidArgumentError: assertion failed: [0] [Op:Assert] name: EagerVariableNameReuse

经过网上查找相关处理方法,分析可能是"使用的时候应该指定未使用的GPU“。没有说具体的解决方法。

我这里的处理的方法是降低Tensorflow-gpu的版本,结果还真解决了这个问题。

原因是我用的版本是tensorflow-gpu-2.3.1 ,稳妥点直接降到 tensorflow-gpu-2.2.0

还真解决了这个问题,给大家参考下。

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐