search

Syn-McJ / TFClassify-Unity

An example of using Tensorflow with Unity for image classification and object detection.
MIT License
167 stars 47 forks source link

is it possible to use tensorflow gpu version? #20

Closed RavenLeeANU closed 5 years ago

RavenLeeANU commented 5 years ago

hi, thanks for your work, I wonder is it possible to use cuda to speed up the inference time. I set up a tensorflow-gpu environment with cuda and cudnn, but I donnot know how to call the gpu operation. Can u give me some hints for that?

Thanks

Syn-McJ commented 5 years ago

Hi @RavenLeeANU,

I assume you set up tensorflow-gpu environment on your working machine? You can train your model using tensorflow-gpu, but it isn't possible to use cuda for inference. Inference happens on the mobile device when you run the sample, not on your machine.

Syn-McJ commented 5 years ago

Hi @RavenLeeANU,

I hope you resolved your problem. I'll close this issue for now, feel free to reopen if you have more questions.