rishizek / tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow
MIT License
286 stars 102 forks source link

How to use GPU to train model #28

Open minjac opened 5 years ago

minjac commented 5 years ago

Hi, I use your project to train a model, but i did not know how to use GPU instead of CPU to train. thanks!

ytliang97 commented 4 years ago

@minjac you should check what version of tensorflow you've download

minjac commented 4 years ago

@minjac you should check what version of tensorflow you've download

thanks