How to train on GPU? - Githubissues

JoshuaPiinRueyPan / ViolenceDetection

Apache License 2.0

152 stars 39 forks source link

How to train on GPU? #35

Open CainXIII opened 4 years ago

CainXIII commented 4 years ago

I tried installing Cuda, Cudnn & Tensorflow for GPU. But when i train the new model, it seem only run on CPU. Are there any changes needed to train on GPU.

maacofficial commented 4 years ago

I solved this problem by installing cuda 9.0