-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
69 lines (53 loc) · 2.09 KB
/
Dockerfile
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# Dockerfile may have following Arguments:
# tag - tag for the Base image, (e.g. 2.9.1 for tensorflow)
# branch - user repository branch to clone (default: master, another option: test)
#
# To build the image:
# $ docker build -t <dockerhub_user>/<dockerhub_repo> --build-arg arg=value .
# or using default args:
# $ docker build -t <dockerhub_user>/<dockerhub_repo> .
#
ARG tag=3.10-bookworm
# Base image, e.g. tensorflow/tensorflow:2.9.1
FROM python:${tag}
LABEL maintainer='Judith Sáinz-Pardo '
LABEL version='0.0.1'
# Federated learning server with flower
# What user branch to clone [!]
ARG branch=main
# Install Ubuntu packages
# - gcc is needed in Pytorch images because deepaas installation might break otherwise (see docs) (it is already installed in tensorflow images)
RUN DEBIAN_FRONTEND=noninteractive apt-get update && \
apt-get install -y --no-install-recommends \
gcc \
git \
curl \
nano \
tini \
&& rm -rf /var/lib/apt/lists/*
RUN wget https://github.com/tsl0922/ttyd/releases/download/1.7.4/ttyd.x86_64 && cp ./ttyd.x86_64 /usr/bin/ttyd && chmod +x /usr/bin/ttyd
# Update python packages
RUN python3 --version && \
pip3 install --no-cache-dir --upgrade pip "setuptools<60.0.0" wheel
# TODO: remove setuptools version requirement when [1] is fixed
# [1]: https://github.com/pypa/setuptools/issues/3301
# Set LANG environment
ENV LANG=C.UTF-8
# Set the working directory
WORKDIR /srv
# Initialization scripts
# deep-start can install JupyterLab or VSCode if requested
RUN git clone https://github.com/ai4os/deep-start /srv/.deep-start && \
ln -s /srv/.deep-start/deep-start.sh /usr/local/bin/deep-start
ENV SHELL=/bin/bash
# Install user app
RUN git clone --depth 1 -b $branch https://github.com/ai4os/ai4os-federated-server && \
cd ai4os-federated-server && \
pip3 install --no-cache-dir -e . && \
cd ..
# Open ports: DEEPaaS (5000), Monitoring (6006), Jupyter (8888)
EXPOSE 5000 6006 8888
# Launch federated-server
#CMD ["python3", "-m", "fedserver.server"]
ENTRYPOINT ["/usr/bin/tini", "--"]
CMD ["ttyd","-p", "6006", "python3", "-m", "fedserver.server"]