Skip to content

Commit

Permalink
ci: addded onnx in docker
Browse files Browse the repository at this point in the history
  • Loading branch information
MrunmayS committed Oct 6, 2023
1 parent d53a131 commit 3d49780
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 4 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/release.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -313,11 +313,11 @@ jobs:

- name: Build package
if: matrix.os != 'ubuntu-20-16-cores'
run: cargo build --release --bin ${{ matrix.file_name }}
run: cargo build --release --bin ${{ matrix.file_name }} --features "onnx"

- name: Build package for ubuntu (with kafka & snowflake)
- name: Build package for ubuntu (with kafka & snowflake & onnx)
if: matrix.os == 'ubuntu-20-16-cores'
run: cargo build --release --bin ${{ matrix.file_name }} --features "kafka snowflake"
run: cargo build --release --bin ${{ matrix.file_name }} --features "kafka snowflake onnx"

- name: Prepare release assets
shell: bash
Expand Down
11 changes: 10 additions & 1 deletion ci/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,9 @@ RUN apt-get update \
odbcinst \
unixodbc \
curl \
unzip
unzip \
cmake \
clang

# INSTALL PROTOBUF
RUN curl -LO https://github.com/protocolbuffers/protobuf/releases/download/v3.18.1/protoc-3.18.1-linux-x86_64.zip
Expand All @@ -18,6 +20,13 @@ RUN curl -LO https://sfc-repo.snowflakecomputing.com/odbc/linux/2.25.7/snowflake
RUN dpkg -i snowflake-odbc-2.25.7.x86_64.deb
RUN rm snowflake-odbc-2.25.7.x86_64.deb

# Install ONNX Runtime
RUN curl -LO https://github.com/microsoft/onnxruntime/releases/download/v1.15.1/onnxruntime-linux-x64-1.15.1.tgz
RUN tar -xf onnxruntime-linux-x64-1.15.1.tgz -C /usr/local
RUN rm onnxruntime-linux-x64-1.15.1.tgz
RUN cp /usr/local/onnxruntime-linux-x64-1.15.1/lib/libonnxruntime.so.1.15.1 /usr/local/lib/
RUN ldconfig

# INSTALL DOZER
RUN echo "Installing dozer binary"

Expand Down

0 comments on commit 3d49780

Please sign in to comment.