From dc8533516289e28170bd861a7d830e4d9fa72512 Mon Sep 17 00:00:00 2001 From: Deyong Zheng Date: Fri, 12 Jun 2020 22:26:10 +0900 Subject: [PATCH] fix import pyyolo error. undefined symbol: _ZGVbN2vv___pow_finite --- Makefile | 2 +- setup.py | 2 ++ setup_gpu.py | 2 ++ 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/Makefile b/Makefile index 8e42705..d098f91 100644 --- a/Makefile +++ b/Makefile @@ -20,7 +20,7 @@ OBJDIR=./obj/ CC=gcc AR=ar NVCC=nvcc -OPTS=-Ofast +OPTS=-Ofast -fopenmp -lm -lmvec COMMON= CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -fPIC CFLAGS+=-I./darknet/src -I./darknet/include diff --git a/setup.py b/setup.py index 626dde4..5b45b36 100644 --- a/setup.py +++ b/setup.py @@ -4,6 +4,8 @@ module = Extension('pyyolo', library_dirs=['.'], libraries=['yolo'], + extra_compile_args=['-fopenmp'], + extra_link_args=['-lgomp', '-lm', '-lmvec'], include_dirs=[numpy.get_include(), './darknet/include'], sources = ['module.c']) diff --git a/setup_gpu.py b/setup_gpu.py index 64003b1..d694d6e 100644 --- a/setup_gpu.py +++ b/setup_gpu.py @@ -4,6 +4,8 @@ module = Extension('pyyolo', library_dirs=['.', '/usr/local/cuda/lib64', '/usr/local/'], libraries=['yolo', 'cuda', 'cudart', 'cublas', 'curand', 'cudnn'], + extra_compile_args=['-fopenmp'], + extra_link_args=['-lgomp', '-lm', '-lmvec'], include_dirs=[numpy.get_include(), './darknet/include'], sources = ['module.c'])