``` cc := g++ name := pro workdir := workspace srcdir := src objdir := objs stdcpp := c++11 cuda_home := /datav/software/anaconda3/lib/python3.9/site-packages/trtpy/trt8cuda112cudnn8 syslib := /datav/software/anaconda3/lib/python3.9/site-packages/trtpy/lib cpp_pkg := /datav/software/anaconda3/lib/python3.9/site-packages/trtpy/cpp-packages cuda_arch := nvcc := $(cuda_home)/bin/nvcc -ccbin=$(cc) # 定义cpp的路径查找和依赖项mk文件 cpp_srcs := $(shell find $(srcdir) -name "*.cpp") cpp_objs := $(cpp_srcs:.cpp=.cpp.o) cpp_objs := $(cpp_objs:$(srcdir)/%=$(objdir)/%) cpp_mk := $(cpp_objs:.cpp.o=.cpp.mk) # 定义cu文件的路径查找和依赖项mk文件 cu_srcs := $(shell find $(srcdir) -name "*.cu") cu_objs := $(cu_srcs:.cu=.cu.o) cu_objs := $(cu_objs:$(srcdir)/%=$(objdir)/%) cu_mk := $(cu_objs:.cu.o=.cu.mk) # 定义opencv和cuda需要用到的库文件 link_cuda := cudart cudnn link_trtpro := link_tensorRT := nvinfer nvinfer_plugin link_opencv := link_sys := stdc++ dl protobuf link_librarys := $(link_cuda) $(link_tensorRT) $(link_sys) $(link_opencv) # 定义头文件路径,请注意斜杠后边不能有空格 # 只需要写路径,不需要写-I include_paths := src \ $(cuda_home)/include/cuda \ $(cuda_home)/include/tensorRT \ $(cpp_pkg)/opencv4.2/include \ $(cuda_home)/include/protobuf # 定义库文件路径,只需要写路径,不需要写-L library_paths := $(cuda_home)/lib64 $(syslib) $(cpp_pkg)/opencv4.2/lib # 把library path给拼接为一个字符串,例如a b c => a:b:c # 然后使得LD_LIBRARY_PATH=a:b:c empty := library_path_export := $(subst $(empty) $(empty),:,$(library_paths)) # 把库路径和头文件路径拼接起来成一个,批量自动加-I、-L、-l run_paths := $(foreach item,$(library_paths),-Wl,-rpath=$(item)) include_paths := $(foreach item,$(include_paths),-I$(item)) library_paths := $(foreach item,$(library_paths),-L$(item)) link_librarys := $(foreach item,$(link_librarys),-l$(item)) # 如果是其他显卡,请修改-gencode=arch=compute_75,code=sm_75为对应显卡的能力 # 显卡对应的号码参考这里:https://developer.nvidia.com/zh-cn/cuda-gpus#compute # 如果是 jetson nano,提示找不到-m64指令,请删掉 -m64选项。不影响结果 cpp_compile_flags := -std=$(stdcpp) -w -g -O0 -m64 -fPIC -fopenmp -pthread cu_compile_flags := -std=$(stdcpp) -w -g -O0 -m64 $(cuda_arch) -Xcompiler "$(cpp_compile_flags)" link_flags := -pthread -fopenmp -Wl,-rpath='$$ORIGIN' cpp_compile_flags += $(include_paths) cu_compile_flags += $(include_paths) link_flags += $(library_paths) $(link_librarys) $(run_paths) # 如果头文件修改了,这里的指令可以让他自动编译依赖的cpp或者cu文件 ifneq ($(MAKECMDGOALS), clean) -include $(cpp_mk) $(cu_mk) endif $(name) : $(workdir)/$(name) all : $(name) run : $(name) @cd $(workdir) && ./$(name) $(run_args) $(workdir)/$(name) : $(cpp_objs) $(cu_objs) @echo Link $@ @mkdir -p $(dir $@) @$(cc) $^ -o $@ $(link_flags) $(objdir)/%.cpp.o : $(srcdir)/%.cpp @echo Compile CXX $< @mkdir -p $(dir $@) @$(cc) -c $< -o $@ $(cpp_compile_flags) $(objdir)/%.cu.o : $(srcdir)/%.cu @echo Compile CUDA $< @mkdir -p $(dir $@) @$(nvcc) -c $< -o $@ $(cu_compile_flags) # 编译cpp依赖项,生成mk文件 $(objdir)/%.cpp.mk : $(srcdir)/%.cpp @echo Compile depends C++ $< @mkdir -p $(dir $@) @$(cc) -M $< -MF $@ -MT $(@:.cpp.mk=.cpp.o) $(cpp_compile_flags) # 编译cu文件的依赖项,生成cumk文件 $(objdir)/%.cu.mk : $(srcdir)/%.cu @echo Compile depends CUDA $< @mkdir -p $(dir $@) @$(nvcc) -M $< -MF $@ -MT $(@:.cu.mk=.cu.o) $(cu_compile_flags) # 定义清理指令 clean : @rm -rf $(objdir) $(workdir)/$(name) $(workdir)/*.trtmodel $(workdir)/demo.onnx # 防止符号被当做文件 .PHONY : clean run $(name) # 导出依赖库路径,使得能够运行起来 export LD_LIBRARY_PATH:=$(library_path_export)```empty := library_path_export := $(subst $(empty) $(empty),:,$(library_paths))解释一下,每一行什么意思,具体一些
时间: 2025-03-07 17:05:05 浏览: 43
```makefile
cc := g++
name := pro
workdir := workspace
srcdir := src
objdir := objs
stdcpp := c++11
cuda_home := /datav/software/anaconda3/lib/python3.9/site-packages/trtpy/trt8cuda112cudnn8
syslib := /datav/software/anaconda3/lib/python3.9/site-packages/trtpy/lib
cpp_pkg := /datav/software/anaconda3/lib/python3.9/site-packages/trtpy/cpp-packages
cuda_arch :=
nvcc := $(cuda_home)/bin/nvcc -ccbin=$(cc)
```
上述部分定义了一系列变量用于配置构建环境。`cc` 指定了使用的C++编译器;`name`, `workdir`, `srcdir`, 和 `objdir` 分别指代项目的名称、工作目录、源代码存放位置以及目标文件(如.o 文件)的输出位置。`stdcpp` 设置了C++标准版本,而 `cuda_home`, `syslib`, 和 `cpp_pkg` 则指向CUDA, 系统库文件和OpenCV等第三方包的具体路径。
```makefile
cpp_srcs := $(shell find $(srcdir) -name "*.cpp")
cpp_objs := $(cpp_srcs:.cpp=.cpp.o)
cpp_objs := $(cpp_objs:$(srcdir)/%=$(objdir)/%)
cpp_mk := $(cpp_objs:.cpp.o=.cpp.mk)
cu_srcs := $(shell find $(srcdir) -name "*.cu")
cu_objs := $(cu_srcs:.cu=.cu.o)
cu_objs := $(cu_objs:$(srcdir)/%=$(objdir)/%)
cu_mk := $(cu_objs:.cu.o=.cu.mk)
```
此段落利用shell命令搜索所有.cpp与.cu结尾的源文件,并转换为目标文件路径及生成相应的依赖关系标记文件(.mk),以供后续自动化管理编译过程中对这些文件的操作[^1]。
```makefile
link_cuda := cudart cudnn
link_tensorRT := nvinfer nvinfer_plugin
link_sys := stdc++ dl protobuf
link_librarys := $(link_cuda) $(link_tensorRT) $(link_sys) $(link_opencv)
```
这部分列出了链接阶段所需的各个库的名字,包括CUDA相关的(cudart,cudnn), TensorRT(nvinfer,nvinfer_plugin),以及其他系统级别的库(stdc++,dl,protobuf)。
```makefile
include_paths := src \
$(cuda_home)/include/cuda \
$(cuda_home)/include/tensorRT \
$(cpp_pkg)/opencv4.2/include \
$(cuda_home)/include/protobuf
library_paths := $(cuda_home)/lib64 $(syslib) $(cpp_pkg)/opencv4.2/lib
```
这两行声明了头文件(`include_paths`)和静态或共享库(`library_paths`)的位置,在实际编译时会被加入到编译参数中去确保能正确找到所需资源。
```makefile
empty :=
library_path_export := $(subst $(empty) $(empty),:,$(library_paths))
run_paths := $(foreach item,$(library_paths),-Wl,-rpath=$(item))
include_paths := $(foreach item,$(include_paths),-I$(item))
library_paths := $(foreach item,$(library_paths),-L$(item))
link_librarys := $(foreach item,$(link_librarys),-l$(item))
```
为了方便动态加载库时能找到正确的路径,此处构造了一个由冒号分隔的字符串形式的库路径列表 (`library_path_export`) 并设置 `-rpath` 参数以便程序执行期解析依赖库。另外也将之前提到过的头文件夹转成了GCC/G++识别的形式 (-I 开头),同样地处理了库路径(-L开头)并最终形成了一组完整的 `-l` 标志来指示要连接哪些外部库。
```makefile
cpp_compile_flags := -std=$(stdcpp) -w -g -O0 -m64 -fPIC -fopenmp -pthread
cu_compile_flags := -std=$(stdcpp) -w -g -O0 -m64 $(cuda_arch) -Xcompiler "$(cpp_compile_flags)"
link_flags := -pthread -fopenmp -Wl,-rpath='$$ORIGIN'
...
clean :
rm -rf $(objdir) $(workdir)/$(name) $(workdir)/*.trtmodel $(workdir)/demo.onnx
.PHONY : clean run $(name)
export LD_LIBRARY_PATH:=$(library_path_export)
```
最后几行为具体的编译标志设定和其他辅助操作说明。设置了针对CPP/CUDA文件的不同编译选项(比如优化级别、线程支持等)。还定义了一些额外的目标标签用来清理中间产物(clean)或者指定某些任务为伪目标(PHONY)防止误判作同名文件存在导致无法触发规则的情况发生。最后一句则是导出必要的环境变量让当前Shell及其子进程中都可以访问到所定义好的库路径信息。
阅读全文