Tensorflow 模型恢复时,设备指定与当前不匹配报错的解决方法

错误内容:nvalidArgumentError (see above for traceback): Cannot assign a device for operation ‘Reshape’: Operation was explicitly assigned to /job:worker/task:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0 ]. Make sure the device specification refers to a valid device.

错误原因:训练时指定的设备与当前环境下的设备不匹配

解决方法:在恢复.meta的时候,添加一个参数clear devices = True即可清除之前配置。
代码样本如下:

saver = tf.train.import_meta_graph('./model.ckpt.meta',clear_devices=True)

我也是查了很久,自己无意中看源码看到了这个参数,才解决了问题,所以分享下,帮助后来人!

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值