MTRNord 2022-10-10 21:29:40 +02:00
parent d2b5ee8927
commit 3341284960
1 changed files with 4 additions and 20 deletions

View File

@ -1,23 +1,4 @@
FROM rust:1.64 as builder
# Install tensorflow
RUN apt-get update && apt-get install -y curl
# See http://bugs.python.org/issue19846
ENV LANG C.UTF-8
RUN apt-get update && apt-get install -y \
python3 \
python3-pip
RUN python3 -m pip --no-cache-dir install --upgrade \
"pip<20.3" \
setuptools
# Some TF tools expect a "python" binary
RUN ln -s $(which python3) /usr/local/bin/python
RUN python3 -m pip install --no-cache-dir tensorflow
FROM rust:1.64
WORKDIR /app
COPY ./crates /app/crates
@ -28,6 +9,9 @@ RUN ls -la target/release/build/tensorflow-sys-*/out
RUN find / -name libtensorflow*
RUN ldd /app/target/release/model_server
RUN find . -type f -name libtensorflow.so.2 -exec cp {} /usr/lib/ \; \
&& find . -type f -name libtensorflow_framework.so.2 -exec cp {} /usr/lib/ \;
ENV MODEL_PATH /app/models/matrix_spam
# Copy the model files to the image
COPY ./models/spam_keras_1664583738.3207538 /app/models/matrix_spam