search

matterport / Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Other
24.63k stars 11.7k forks source link

How can I choose GPU id in a multi-gpu computer? #814

Open Zwenjay opened 6 years ago

Zwenjay commented 6 years ago

I think we need such a config variable

rbavery commented 6 years ago

See this notebook, you can set the device with tf.device()

https://github.com/matterport/Mask_RCNN/blob/master/samples/nucleus/inspect_nucleus_model.ipynb

Ryan Avery Graduate Student, WAVES Lab Department of Geography University of California, Santa Barbara

http://caylor.eri.ucsb.edu/people/avery/ http://caylor.eri.ucsb.edu/people/avery/

On Thu, Jul 26, 2018 at 4:36 AM, Zwenjay notifications@github.com wrote:

I think we need such a config variable

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/matterport/Mask_RCNN/issues/814, or mute the thread https://github.com/notifications/unsubscribe-auth/AVOkCV1N9lAW20q-mXKrLn89Ke-0tH07ks5uKX9xgaJpZM4VhZoM .