search

digitalbrain79 / pyyolo

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

Assertion '0' failed #15

Closed zspasztori closed 7 years ago

zspasztori commented 7 years ago

Hi, I make with GPU=1, CUDNN=1, but when I run the example I get:

----- test original C using a file CUDA Error: invalid device function python: ./darknet/src/cuda.c:36: check_error: Assertion `0' failed. Aborted (core dumped)

The original Darknet works fine for me, when I set the ARCH to --gpu-architecture=compute_35 --gpu-code=compute_35. Unfortunatelly here this does not seem to work.

zspasztori commented 7 years ago

The cuda library path was not set well in the setup_gpu.py file. For some reason my cuda library is at cuda-8.0