search

dpressel / rude-carnie

Age detection in Tensorflow
937 stars 341 forks source link

not working on gpu ;( #24

Closed orlov-alexander closed 7 years ago

orlov-alexander commented 7 years ago

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 commented 7 years ago

I believe the change below fixes this issue:

https://github.com/dpressel/rude-carnie/commit/8a8c1231fd1fb99a15dd4b4d6bc735eef63f52b9

It works for me.

See also:

https://github.com/tensorflow/tensorflow/issues/2292 https://github.com/tensorflow/tensorflow/issues/2285