Tensorflow-gpu tf.keras.Sequential 模型建立时的错误

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

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值