How to use GPU training

GeorgeSeif / Semantic-Segmentation-Suite

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!

2.51k stars 880 forks source link

Closed upuil closed 5 years ago

upuil commented 5 years ago

I change the line with tf.device('/cpu:0'): to with tf.device('/gpu:0'): in train.pydoesn't work.

upuil commented 5 years ago

It works ok, because of my worry configure