Darknet framework fails to start with GPU acceleration using CUDA

Question

I was trying to run Darknet with GPU acceleration using CUDA API. So I followed instructions from here, changed GPU=1 in Makefile and started make. When I'm trying to run test it fails due to CUDA Error.

./darknet yolo test cfg/yolo.cfg yolo.weights data/dog.jpg CUDA Error: unknown error darknet: ./src/cuda.c:21: check_error: Assertion `0' failed.

I'm using Ubuntu 14.04, CUDA 7.5 and my NVIDIA-SMI 352.93 and Driver Version: 352.93 on Titan X I'm pretty sure that my CUDA works fine and driver's version is up to date, because I'm using it to accelerate Caffe. My guess is that Darknet cannot locate CUDA directory.

Can anyone help me with that issue?

Answer 1

you should modify the value of configure parameter 'ARCH'. the default value is --gpu-architecture=compute_52, --gpu-code=compute_52. my setting is --gpu-architecture=compute_30, --gpu-code=compute_30, and it works. it depends on your actual gpu architecture. more detail is in cuda toolkit documentation.