search

youchangxin / YOLOv4_tensorflow2

YOLOv4 Implemented in Tensorflow 2.1，which supports training on custom dataset.
MIT License
25 stars 5 forks source link

Not support gpu? #1

Closed Ikhwansong closed 4 years ago

Ikhwansong commented 4 years ago

hi, Thank for your sharing code.

I have a one question : Do not this code support gpu version by tensorflow?

youchangxin commented 4 years ago

It certainly supports TensorFlow GPU.

Ikhwansong commented 4 years ago

Because of absence of cudnn, cudatoolkit. it didn't work I just have checked it and download cudnn&cudatoolkit by anaconda command then I could see it works! Thanks