Adding framework-specifc dockerfiles

Dockerfile

-FROM ubuntu:20.04
+FROM ubuntu:20.04 as builder
 # Define workdir
 WORKDIR /experiment
 # Get basics
@@ -14,9 +14,11 @@ ENV VIRTUAL_ENV=/opt/declearn
 RUN python3 -m venv $VIRTUAL_ENV
 ENV PATH="$VIRTUAL_ENV/bin:$PATH"
 # Install declearn
-RUN pip install declearn[all]
-# Create env variables
-# Run script
+RUN pip install declearn[grpc,websockets]
+# Use multi-stage build to save space
+FROM alpine
+COPY --from=builder /opt/declearn /opt/declearn
+# Define and run script
 ENV script="experiment/run.sh"
 SHELL ["/bin/bash", "-c"]
-CMD chmod u+x ${script} && ${script}
+CMD chmod u+x ${script} && ${script}
\ No newline at end of file

haiku.Dockerfile

+FROM ubuntu:20.04 as builder
+# Define workdir
+WORKDIR /experiment
+# Get basics
+RUN apt-get update && \
+    apt-get install -y python3 \
+    python3-pip \
+    git \
+    gcc \
+    python3-venv \
+    bash
+# Set up venv
+ENV VIRTUAL_ENV=/opt/declearn
+RUN python3 -m venv $VIRTUAL_ENV
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+# Install declearn
+RUN pip install declearn[haiku,grpc,websockets]
+# Use multi-stage build to save space
+FROM alpine
+COPY --from=builder /opt/declearn /opt/declearn
+# Define and run script
+ENV script="experiment/run.sh"
+SHELL ["/bin/bash", "-c"]
+CMD chmod u+x ${script} && ${script}

tensorflow.Dockerfile

+FROM ubuntu:20.04 as builder
+# Define workdir
+WORKDIR /experiment
+# Get basics
+RUN apt-get update && \
+    apt-get install -y python3 \
+    python3-pip \
+    git \
+    gcc \
+    python3-venv \
+    bash
+# Set up venv
+ENV VIRTUAL_ENV=/opt/declearn
+RUN python3 -m venv $VIRTUAL_ENV
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+# Install declearn
+RUN pip install declearn[tensorflow,grpc,websockets]
+# Use multi-stage build to save space
+FROM alpine
+COPY --from=builder /opt/declearn /opt/declearn
+# Define and run script
+ENV script="experiment/run.sh"
+SHELL ["/bin/bash", "-c"]
+CMD chmod u+x ${script} && ${script}

torch.Dockerfile

+FROM ubuntu:20.04 as builder
+# Define workdir
+WORKDIR /experiment
+# Get basics
+RUN apt-get update && \
+    apt-get install -y python3 \
+    python3-pip \
+    git \
+    gcc \
+    python3-venv \
+    bash
+# Set up venv
+ENV VIRTUAL_ENV=/opt/declearn
+RUN python3 -m venv $VIRTUAL_ENV
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+# Install declearn
+RUN pip install declearn[torch,grpc,websockets]
+# Use multi-stage build to save space
+FROM alpine
+COPY --from=builder /opt/declearn /opt/declearn
+# Define and run script
+ENV script="experiment/run.sh"
+SHELL ["/bin/bash", "-c"]
+CMD chmod u+x ${script} && ${script}