UNet training Docker (#31)

Add Dockerfile for U-Net training environment
DIAGNijmegen · Dec 21, 2022 · 6865613 · 6865613
1 parent 3a9cfa3
commit 6865613
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/picai_baseline/unet/Dockerfile b/src/picai_baseline/unet/Dockerfile
@@ -0,0 +1,7 @@
+FROM joeranbosma/picai_nnunet:latest
+
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install -r /tmp/requirements.txt
+RUN pip install git+https://github.com/DIAGNijmegen/picai_baseline
+
+ENTRYPOINT $0 $@