Run system on tensorflow-gpu

OlafenwaMoses / Traffic-Net

A dataset of traffic, fire and accident images for training deep learning models.

https://DeepQuestAI.com

MIT License

151 stars 50 forks source link

Run system on tensorflow-gpu #4

Open atzoriandrea opened 4 years ago

atzoriandrea commented 4 years ago

Hi. I'm trying to run this system on tensorflow-gpu, but it raises some exceptions with cudnn (tf-gpu 1.14; CUDA 10.0; cudnn 7.4.2). When i run it on CPU, instead, no problem appears. So, it is possible to run it on GPU or not?