Tensorflow: Can I export a gpu-trained model to a cpu-only machine for serving/evaluation?

Created on 16 Jul 2016  ·  3Comments  ·  Source: tensorflow/tensorflow

I trained a model on a GPU machine.
When l load such model on a cpu-only machine, I get the error below.

Thank you for any suggestions!

awaiting tensorflower
Source
picture 3rduncle

Most helpful comment

Also, in the future, StackOverflow is the right place to ask these questions.

picture vrv on 18 Jul 2016
👍2

All 3 comments

@vrv Could you take a look at this please? Do we have such support now? Thanks.

jmchen-g picture jmchen-g on 18 Jul 2016

Yup, you can pass tf.ConfigProto(allow_soft_placement=True) to the tf.Session if you really want to ignore device placement directives in the graph. I'm not a huge fan of using this option (I'd rather the graph be rewritten to strip out the device fields explicitly), but that should work.

vrv picture vrv on 18 Jul 2016
👍2

Also, in the future, StackOverflow is the right place to ask these questions.

vrv picture vrv on 18 Jul 2016
👍2
Was this page helpful?
0 / 5 - 0 ratings

Related issues

waleedka picture waleedka  ·  3Comments
untom picture untom  ·  3Comments
Billy4195 picture Billy4195  ·  3Comments
as1ndu picture as1ndu  ·  3Comments
jacobma-create picture jacobma-create  ·  3Comments