diff --git a/samples/docker/serve-custom-sample/Dockerfile b/samples/docker/serve-custom-sample/Dockerfile index 5fe35659b..a7013d943 100644 --- a/samples/docker/serve-custom-sample/Dockerfile +++ b/samples/docker/serve-custom-sample/Dockerfile @@ -1,5 +1,5 @@ FROM happy365/fast-style-transfer-build as builder -FROM tensorflow/tensorflow:1.5.0-gpu +FROM tensorflow/tensorflow:2.16.1-gpu RUN mv /etc/apt/sources.list.d /etc/apt/sources.list.d.bak && \ apt-get update RUN apt-get install apt-transport-https -y && \