From 9805bb1aa52757705555363cac9435f81a3d48b7 Mon Sep 17 00:00:00 2001
From: JoelH96 <41146991+JoelH96@users.noreply.github.com>
Date: Wed, 6 Nov 2019 16:19:15 +0000
Subject: [PATCH] Updated GPU wrapper to uninstall TF without GPU (#1011)

* Updated GPU wrapper to uninstall TF without GPU

* Updated GPU wrapper

* Updated Python-GPU Wrapper
---
 wrappers/s2i/python-gpu/python-gpu/Dockerfile | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/wrappers/s2i/python-gpu/python-gpu/Dockerfile b/wrappers/s2i/python-gpu/python-gpu/Dockerfile
index cd65eedeee..2e186f7491 100644
--- 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