search

bendangnuksung / mrcnn_serving_ready

🛠 Mask R-CNN Keras to Tensorflow and TFX models + Serving models using TFX GRPC & RESTAPI
https://github.com/matterport/Mask_RCNN
MIT License
103 stars 35 forks source link

InvalidArgumentError: Cannot assign a device for operation ROI/rpn_non_max_suppression/NonMaxSuppressionV3 #13

Open K-M-Ibrahim-Khalilullah opened 5 years ago

K-M-Ibrahim-Khalilullah commented 5 years ago

When I run your code (a little modified for config) it gives the following error:

InvalidArgumentError: Cannot assign a device for operation ROI/rpn_non_max_suppression/NonMaxSuppressionV3: Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available. Registered kernels: device='CPU'; T in [DT_FLOAT] device='CPU'; T in [DT_HALF]

Maybe the problem arise due these lines:

sess = tf.Session() K.set_session(sess)

How can I solved?