Updated GPU wrapper to uninstall TF without GPU (#1011)

* Updated GPU wrapper to uninstall TF without GPU * Updated GPU wrapper * Updated Python-GPU Wrapper
SeldonIO · Nov 6, 2019 · 9805bb1 · 9805bb1
1 parent 620988a
commit 9805bb1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/wrappers/s2i/python-gpu/python-gpu/Dockerfile b/wrappers/s2i/python-gpu/python-gpu/Dockerfile
@@ -17,7 +17,11 @@ WORKDIR /microservice
 COPY ./s2i/bin/ /s2i/bin
 
 # keep install of seldon-core after the COPY to force re-build of layer
-RUN pip3 install tensorflow-gpu==1.14.0
 RUN pip3 install seldon-core
+RUN pip3 install tensorflow-gpu
+
+# Numpy version 1.16.1 required for tf-gpu
+RUN pip3 install numpy==1.16.1
+
 
 EXPOSE 5000