investigations and dockerfile to finallise PETRIC installation

SyneRBI · Oct 17, 2024 · c8f242f · c8f242f
1 parent 4e8dd56
commit c8f242f
Show file tree

Hide file tree

Showing 4 changed files with 284 additions and 257 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,12 @@
+# syntax=docker/dockerfile:1
+ARG BASE_CONTAINER=ghcr.io/synerbi/sirf:edge-gpu
+FROM ${BASE_CONTAINER} AS base
+
+RUN conda install -y monai tensorboard tensorboardx jupytext cudatoolkit=11.8
+# monai installs pytorch (CPU), so remove it
+RUN pip uninstall -y torch
+# last to support cu118
+RUN pip install tensorflow[and-cuda]==2.14  
+RUN pip install torch --index-url https://download.pytorch.org/whl/cu118
+RUN pip install git+https://github.com/TomographicImaging/Hackathon-000-Stochastic-QualityMetrics
+RUN pip install pysnooper
diff --git a/main.py b/main.py
diff --git a/main.py b/main.py
@@ -0,0 +1 @@
+main_submission.py