search

microsoft / onnxruntime-extensions

onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime
MIT License
300 stars 80 forks source link

Make ort extensions build successfully on Linux GPU #609

Closed mszhanyi closed 7 months ago

mszhanyi commented 7 months ago

How to test locally: export CUDA_PATH=/usr/local/cuda-11.8 sh ./build.sh -DOCOS_ENABLE_CTEST=ON -DOCOS_ONNXRUNTIME_VERSION={ORT_VERSION} -DOCOS_USE_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=86 -DONNXRUNTIME_PKG_DIR={ORT_RELEASE_DIR}

Note:

  1. Since std::filesystem is used in unit test, gcc version should be bigger than 9.0
  2. CI pipeline will be enabled in next PR.