search

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 399 forks source link

Lumi GPU support? #200

Closed akorez closed 6 years ago

akorez commented 6 years ago

How do I know if I'm using the GPU? Predict and train are doing things very slowly? Is there a parameter to use the GPU? Thank you...

nagitsu commented 6 years ago

In order to use the GPU you must have installed the tensorflow-gpu instead of the tensorflow dependency. While running, you can check the GPU usage using the nvidia-smi command, assuming you have an Nvidia GPU.