适配linux cuda

onnxruntime-gpu/linux/OnnxRuntimeConfig.cmake

 set(OnnxRuntime_INCLUDE_DIRS "${CMAKE_CURRENT_LIST_DIR}/include") 
 include_directories(${OnnxRuntime_INCLUDE_DIRS}) 
 link_directories(${CMAKE_CURRENT_LIST_DIR}/lib) 
-set(OnnxRuntime_LIBS onnxruntime onnxruntime_providers_cuda onnxruntime_providers_shared onnxruntime_providers_tensorrt) 
+set(OnnxRuntime_LIBS onnxruntime)

onnxruntime-gpu/windows-x64/OnnxRuntimeConfig.cmake

 set(OnnxRuntime_INCLUDE_DIRS "${CMAKE_CURRENT_LIST_DIR}/include") 
 include_directories(${OnnxRuntime_INCLUDE_DIRS}) 
 link_directories(${CMAKE_CURRENT_LIST_DIR}/lib) 
-set(OnnxRuntime_LIBS onnxruntime onnxruntime_providers_cuda onnxruntime_providers_shared onnxruntime_providers_tensorrt) 
+set(OnnxRuntime_LIBS onnxruntime)

run-benchmark.sh

 #!/usr/bin/env bash
 
 function PrepareVar(){
-EXE_PATH=${sysOS}-BIN
+echo "onnxruntime: 1)CPU(默认), 2)GPU(cuda)"
+read -p "" ONNX_TYPE
+if [ $ONNX_TYPE == 1 ]; then
+  ONNX_TYPE="CPU"
+  GPU_INDEX=-1
+elif [ $ONNX_TYPE == 2 ]; then
+  ONNX_TYPE="CUDA"
+  GPU_INDEX=0
+else
+  echo -e "输入错误！Input Error!"
+fi
+EXE_PATH=${sysOS}-${ONNX_TYPE}-BIN
 }
 
 sysOS=`uname -s`
@@ -45,4 +56,5 @@ fi
 --unClipRatio 1.6 \
 --doAngle 1 \
 --mostAngle 1 \
+--GPU $GPU_INDEX \
 --loopCount $LOOP_COUNT
\ No newline at end of file

run-test.sh

 #!/usr/bin/env bash
 
 function PrepareVar(){
-EXE_PATH=${sysOS}-BIN
+echo "onnxruntime: 1)CPU(默认), 2)GPU(cuda)"
+read -p "" ONNX_TYPE
+if [ $ONNX_TYPE == 1 ]; then
+  ONNX_TYPE="CPU"
+  GPU_INDEX=-1
+elif [ $ONNX_TYPE == 2 ]; then
+  ONNX_TYPE="CUDA"
+  GPU_INDEX=0
+else
+  echo -e "输入错误！Input Error!"
+fi
+EXE_PATH=${sysOS}-${ONNX_TYPE}-BIN
 }
 
 sysOS=`uname -s`
@@ -42,4 +53,5 @@ fi
 --boxThresh 0.3 \
 --unClipRatio 1.6 \
 --doAngle 1 \
---mostAngle 1
\ No newline at end of file
+--mostAngle 1 \
+--GPU $GPU_INDEX
\ No newline at end of file

valgrind-massif.sh

@@ -18,6 +18,9 @@ echo "找不到待识别的目标图片：${TARGET_IMG}，请打开本文件并
 exit
 fi
 
+sysOS=`uname -s`
+EXE_PATH=${sysOS}-CPU-BIN
+
 ##### run test on MacOS or Linux
 valgrind --tool=massif --pages-as-heap=yes \
 ./${EXE_PATH}/RapidOcrOnnx --models models \

valgrind-memcheck.sh

@@ -19,7 +19,7 @@ exit
 fi
 
 sysOS=`uname -s`
-EXE_PATH=${sysOS}-BIN
+EXE_PATH=${sysOS}-CPU-BIN
 
 ##### run test on MacOS or Linux
 valgrind --tool=memcheck --leak-check=full --leak-resolution=med --track-origins=yes --vgdb=no --log-file=valgrind-memcheck.txt \
@@ -36,4 +36,5 @@ valgrind --tool=memcheck --leak-check=full --leak-resolution=med --track-origins
 --boxThresh 0.3 \
 --unClipRatio 1.6 \
 --doAngle 1 \
---mostAngle 1
+--mostAngle 1 \
+--GPU -1