search

digitalbrain79 / pyyolo

Simple python wrapper for YOLO.
126 stars 65 forks source link

In the makefile, do not set CUDNN=1 when using GPU #17

Open kennetms opened 7 years ago

kennetms commented 7 years ago

Just a warning, if you want to use pyyolo with GPU support, make sure CUDNN=0 in the makefile.

Also, make sure that line 13 in the makefile is commented out.

Otherwise, pyyolo works well, even on the latest version of darknet code, as long as you add lstm_layer.o to the makefile objects.

Also, thank you for making pyyolo.

digitalbrain79 commented 7 years ago

Thanks for good information!