更改部署GPU时的数量

Question

我在2-GPU机器上调整了Inception v3。 现在我想用1 GPU上运行的另一台机器上训练的模型，但我得到了这样的错误：

无法分配装置到节点tower_1 /梯度/ tower_1/CONV0/BatchNorm /瞬间/ fully_statistics/Sub_grad/BroadcastGradientArgs'：无法满足显式设备规范'/ device：GPU：1'，因为在此过程中没有注册符合该规范的设备;可用设备：/ job：localhost/replica：0 /任务：0/cpu：0，/ job：localhost/replica：0 /任务：0/gpu：0

看起来模型需要2 -GPU环境就像它被训练的那样。我可以将此模型转换为仅使用1个GPU吗？

Answer 1

我改变了两件事，它的工作。

1. 在会议allow_soft_placement选项打开

   ：

   config = tf.ConfigProto(allow_soft_placement=True) 
   sess = tf.Session(config=config) 
   

2. 重新命名模型文件名从model.ckpt-50000.data-00000-of-00001到model.ckpt-50000