标签： tensorflow gpu imagenet
我使用slim定义的Inception V3模型来预测我自己的图像。从检查点文件恢复模型变量。当我运行它时，我发现没有使用GPU。我设置了tf.device('/gpu:0')，但它不起作用。所以，我很困惑，为什么训练/重新训练模型可以在GPU上工作，但评估不能？我只是按照inception_eval.py中的代码。 enter image description here
tf.device('/gpu:0')
