feat(Dockerfile): update to tensorflow 2.9.1 and set WORKDIR to /home…

…/data by default
MedVisBonn · Jul 7, 2022 · 2709a68 · 2709a68
1 parent 61d2913
commit 2709a68
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,11 +1,13 @@
-FROM tensorflow/tensorflow:2.7.0-gpu AS gpu
+FROM tensorflow/tensorflow:2.9.1-gpu AS gpu
 COPY ./dist /dist
 RUN pip install --upgrade pip && pip install --no-cache-dir /dist/*.whl && rm -R /dist
 RUN mkdir -p /home/.config/ && chmod -R a+rwx /home/
 ENV MPLCONFIGDIR "/home/.config"
+WORKDIR /home/data
 
-FROM tensorflow/tensorflow:2.7.0 AS cpu
+FROM tensorflow/tensorflow:2.9.1 AS cpu
 COPY ./dist /dist
 RUN pip install --upgrade pip && pip install --no-cache-dir /dist/*.whl && rm -R /dist
 RUN mkdir -p /home/.config/ && chmod -R a+rwx /home/
 ENV MPLCONFIGDIR "/home/.config"
+WORKDIR /home/data