Tensorflow-SegNet icon indicating copy to clipboard operation
Tensorflow-SegNet copied to clipboard

Published 20 hours ago • verified publisher icon tkuanlun350

Can it use cpu only?

Open AllanZuo opened this issue 7 years ago • 1 comments
trafficstars

My device doesn't support GPU.Can it be changed to support cpu?

AllanZuo avatar Jun 04 '18 12:06 AllanZuo

You can change utils.py, line 58 with tf.device('/gpu:0'): to with tf.device('/cpu:0'):

dungba88 avatar Jun 10 '18 09:06 dungba88