- 在onnxruntime的github下载对应的源码,版本不要太新,下载时要加上git --recursive,否则会导致有些包缺失。
git clone --branch v1.5.2 --recursive https://gitee.com/lee-zq/onnxruntime.git
- 根据runtime的版本安装对应的cuda和cudnn。版本不对应会报错
- 设置环境变量为所用的cuda:
export LD_LIBRARY_PATH=/data1/env/cuda-11.4/lib64:$LD_LIBRARY_PATH
export PATH=$PATH:/data1/env/cuda-11.4/bin:$PATH
运行runtime文件夹下的make.sh,
./build.sh --skip_tests --use_cuda --config Release --build_shared_lib --parallel --cuda_home /usr/local/cuda-11.3 --cudnn_home /usr/local/cuda-11.3
在CMakeList.txt中添加对应的依赖:
set(LIBONNX_PATH "/home/onnx/onnxruntime/build/Linux/RelWithDebInfo")
file(GLOB ONNX_LIBS "${LIBONNX_PATH}/*.so")
include_directories(
include
)