-
Notifications
You must be signed in to change notification settings - Fork 9
/
Dockerfile-cpu
26 lines (19 loc) · 917 Bytes
/
Dockerfile-cpu
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
FROM pytorch/torchserve:0.8.2-cpu
USER root
RUN apt update \
&& apt install -y unzip git curl \
&& curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip" \
&& unzip awscliv2.zip \
&& ./aws/install
RUN pip install "onnxruntime==1.14.1" "onnx==1.13.1" "git+https://github.com/facebookresearch/segment-anything.git" "shapely" "pyproj" "rasterio"
COPY ./deployment/config_decode.properties /home/model-server/config.properties
COPY ./deployment/append_memory_setting.sh /home/model-server/append_memory_setting.sh
RUN chmod +x /home/model-server/append_memory_setting.sh \
&& /home/model-server/append_memory_setting.sh
COPY start ./
WORKDIR /home/model-server
COPY dist/sam_serve-1.0.0-py3-none-any.whl /home/model-server
RUN pip install /home/model-server/sam_serve-1.0.0-py3-none-any.whl
ENV TEMP=/home/model-server/tmp
ENV ENABLE_TORCH_PROFILER=FALSE
CMD ./start