1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- SONAME = libtensorrtpro.so
- SOLINK = $(SONAME).0.0
- SOTARGET = $(SONAME).0.0.0
- cuda_arch = -gencode=arch=compute_75,code=sm_75 # NVIDIA GeForce RTX 2070
- SOLIB_OBJS = application/yolocrowd.o\
- tensorrt/cuda_tools.o tensorrt/ilogger.o \
- tensorrt/trt_infer.o tensorrt/trt_tensor.o
- SOLIB_CU_OBJS = application/yolocrowd_decode.o tensorrt/preprocess_kernel.o
- INCLUDE = -I../include/application \
- -I../include/tensorrt \
- -I../../../../../links/common/include \
- -I/usr/local/include/opencv4 \
- -I/usr/local/cuda/include
- CC = g++
- NVCC = nvcc
- RM = rm -f
- LN = ln -sf
- CXXFLAGS = -std=gnu++11 -g -O0 -Wall -fPIC -DLINUX -D__STDC_CONSTANT_MACROS
- NVCCFLAGS = -std=c++11 -g -w -O0 -Xcompiler "$(CXXFLAGS)" $(cuda_arch)
- LDPATH = -L../../../../../links/common/libs -L/usr/local/cuda/lib64 -L/home/flechazo/tools/TensorRT-8.5.3.1/lib
- LDFLAGS = -lpthread -ldl -lcommon\
- -lopencv_core -lopencv_imgproc -lopencv_imgcodecs \
- -lnvinfer -lnvinfer_plugin\
- -lcuda -lcudart
- .SUFFIXES: .cu .cpp .h .o
- .cpp.o:
- $(CC) $(CXXFLAGS) $(INCLUDE) -o $@ -c $<
- .cu.o:
- $(NVCC) $(NVCCFLAGS) $(INCLUDE) -o $@ -c $<
- $(SOTARGET): $(SOLIB_OBJS) $(SOLIB_CU_OBJS)
- $(CC) -shared -Wl,-Bsymbolic -o $@ $^ $(LDPATH) $(LDFLAGS)
- $(LN) $(SOTARGET) $(SOLINK)
- $(LN) $(SOLINK) $(SONAME)
- all: $(SOTARGET)
- clean:
- $(RM) $(SONAME) $(SOLIB_OBJS) $(SOLIB_CU_OBJS) $(SOLINK) $(SOTARGET)
- .PHONY: clean all
|