How to run your program on GPU

pse-ecn / pose-sensitive-embedding

Pose Sensitive Embedding for Person Re-Identification (PSE)

112 stars 36 forks source link

How to run your program on GPU #2

Closed ouyangpingbu closed 6 years ago

ouyangpingbu commented 6 years ago

hi, I am trying to train a model using your trainer_preid.py. After i start the training program every time, it only runs on CPU. How can I turn it on GPU to run?

andreas-eberle commented 6 years ago

Did you install tensorflow-gpu? If you do, it should automatically run on GPU.