chagne the LE ml matesr image for use with mri train

PHT-Medic · Oct 4, 2023 · 2f1621e · 2f1621e
1 parent 2607dd7
commit 2f1621e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/data/leuko-expert/ml/Dockerfile b/data/leuko-expert/ml/Dockerfile
@@ -1,8 +1,8 @@
-FROM python:3.9
+FROM nvcr.io/nvidia/tensorflow:23.06-tf2-py3
 
 RUN apt-get update -yqq && \
     apt-get dist-upgrade -yqq && \
-    apt-get install -yqq git
+    pip3 install git+https://github.com/PHT-Medic/train-container-library.git
 
 COPY requirements.txt /tmp/requirements.txt
 

diff --git a/data/leuko-expert/ml/requirements.txt b/data/leuko-expert/ml/requirements.txt
@@ -3,8 +3,8 @@ numpy
 pandas
 scikit-learn
 scikit-image
-torch
-torchvision
+nibabel
+scipy
 pydicom
 matplotlib
 tqdm