update CMakeList.txt

Author: pcb9382

CMakeLists.txt

@@ -4,8 +4,14 @@ cmake_minimum_required (VERSION 2.6)
 project (FaceAlgorithmDemo)
 
 # tensorrt 目录
-set(TensorRT_INCLUDE "/home/pcb/TensorRT-8.2.5.1/include" CACHE INTERNAL "TensorRT Library include location")
-set(TensorRT_LIB "/home/pcb/TensorRT-8.2.5.1/lib" CACHE INTERNAL "TensorRT Library lib location")
+# tensorrt path
+if (CMAKE_SYSTEM_PROCESSOR MATCHES "aarch64")
+    set(TensorRT_INCLUDE "/usr/include/aarch64-linux-gnu" CACHE INTERNAL "TensorRT Library include location")
+    set(TensorRT_LIB "/usr/lib/aarch64-linux-gnu" CACHE INTERNAL "TensorRT Library lib location")
+else()
+    set(TensorRT_INCLUDE "/home/pcb/TensorRT-8.2.5.1/targets/x86_64-linux-gnu/include" CACHE INTERNAL "TensorRT Library include location")
+    set(TensorRT_LIB "/home/pcb/TensorRT-8.2.5.1/targets/x86_64-linux-gnu/lib" CACHE INTERNAL "TensorRT Library lib location")
+endif()
 
 #添加子目录
 add_subdirectory("FaceAlgorithm")