updates for TRT10

dusty-nv · Oct 16, 2024 · c038530 · c038530
1 parent e92f8d2
commit c038530
Show file tree

Hide file tree

Showing 2 changed files with 165 additions and 28 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -191,12 +191,18 @@ add_subdirectory(python)
 
 
 # set linker options
-target_link_libraries(jetson-inference jetson-utils nvinfer nvinfer_plugin nvcaffe_parser)
+target_link_libraries(jetson-inference jetson-utils nvinfer nvinfer_plugin)
 
 if(CUDA_VERSION_MAJOR GREATER 9)
 	target_link_libraries(jetson-inference nvonnxparser)
 endif()
 
+if(CUDA_VERSION VERSION_LESS "12.6")
+	target_link_libraries(jetson-inference nvcaffe_parser)
+else()
+    CUDA_ADD_CUBLAS_TO_TARGET(jetson-inference)
+endif()
+
 if(HAS_OPENCV) 
    message("-- linking jetson-inference with OpenCV " ${OpenCV_VERSION})
    target_link_libraries(jetson-inference opencv_core opencv_calib3d)