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