ENH: Dockerfile for flyby and vtk8 no X

DCBIA-OrthoLab · Mar 29, 2021 · 814897b · 814897b
1 parent 389cd81
commit 814897b
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/src/kubernetes/Dockerfile b/src/kubernetes/Dockerfile
@@ -1,10 +1,22 @@
-FROM tensorflow/tensorflow:2.4.0
+FROM tensorflow/tensorflow:2.4.1-gpu
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
 	git \
 	libx11-6 \
-	libgl1
+	libgl1 \
+	libopengl0 \
+	libegl1
 
-RUN pip install itk vtk pandas
-RUN git clone https://github.com/DCBIA-OrthoLab/fly-by-cnn.git
+COPY vtk8.2.0.zip /usr/local/lib/python3.6/dist-packages/
+
+WORKDIR /usr/local/lib/python3.6/dist-packages/
+RUN unzip vtk8.2.0.zip
+
+RUN pip install itk sklearn pandas matplotlib
+
+WORKDIR /app
+RUN git clone https://github.com/DCBIA-OrthoLab/fly-by-cnn.git
+
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.6/dist-packages/vtkmodules/
+ENV MESA_GL_VERSION_OVERRIDE=3.2