not working on gpu ;(

[orlov-alexander]

Hello!

I'm trying to use your model on gpu. On cpu it works flawless, but then I try to use it on gpu - it doesn't work.

Error traceback, in short:

InvalidArgumentError (see above for traceback): Cannot assign a device to node 'save/SaveV2': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.

[[Node: save/SaveV2 = SaveV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT.....
_device="/device:GPU:0"](save/Const, save/SaveV2/tensor_names, save/SaveV2/shape_and_slices, InceptionV3/Conv2d_1a_3x3/BatchNorm/beta, InceptionV3/Conv2d_1a_3x3/BatchNorm/moving_mean, InceptionV3/Conv2d_1a_3x3/BatchNorm/moving_variance, InceptionV3/Conv2d_1a_3x3/weights,

Full traceback and installed python libraries can be seen at https://pastebin.com/aR5CX3eG

I'm using Ubuntu 16.04.1 LTS on a p2.xlarge AWS instance with Tesla K80.

Any ideas why it doesn't work?

[dpressel]

I believe the change below fixes this issue:

8a8c123

It works for me.

See also:

tensorflow/tensorflow#2292
tensorflow/tensorflow#2285