trenden commited on
Commit
fc36823
·
verified ·
1 Parent(s): 942fb6e

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +33 -0
Dockerfile ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10.14-bookworm
2
+
3
+ ARG USER_UID=10002
4
+ ARG USER_GID=$USER_UID
5
+ ARG USERNAME=modelapi
6
+
7
+ RUN groupadd --gid $USER_GID $USERNAME \
8
+ && useradd --uid $USER_UID --gid $USER_GID -m $USERNAME
9
+
10
+ # Copy required files
11
+ RUN mkdir -p /modelapi && mkdir -p /home/$USERNAME/.modelapi
12
+ COPY app /modelapi/app
13
+ COPY sgmse /modelapi/sgmse
14
+ COPY pyproject.toml /modelapi/pyproject.toml
15
+
16
+ ENV CUDA_HOME=/usr/local/cuda-12.6
17
+
18
+ # Setup permissions
19
+ RUN chown -R $USER_UID:$USER_GID /modelapi \
20
+ && chown -R $USER_UID:$USER_GID /home/$USERNAME/.modelapi \
21
+ && chown -R $USER_UID:$USER_GID /home/$USERNAME \
22
+ && chmod -R 755 /home/$USERNAME \
23
+ && chmod -R 755 /modelapi \
24
+ && chmod -R 755 /home/$USERNAME/.modelapi
25
+
26
+ # Change to the user and do subnet installation
27
+ USER $USERNAME
28
+
29
+ RUN /bin/bash -c "python3 -m venv /modelapi/.venv && source /modelapi/.venv/bin/activate && pip3 install -e /modelapi/."
30
+
31
+ EXPOSE 6500
32
+
33
+ CMD ["/bin/bash", "-c", "source /modelapi/.venv/bin/activate && python3 /modelapi/app/run.py"]