From bd6f2b2235779e4705e61d06306b30c433f8840b Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Tue, 15 Feb 2022 08:43:33 -0800 Subject: [PATCH] Update build_wheel.Dockerfile --- tools/docker/build_wheel.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/docker/build_wheel.Dockerfile b/tools/docker/build_wheel.Dockerfile index c4eca2eb88..cf186fddff 100644 --- a/tools/docker/build_wheel.Dockerfile +++ b/tools/docker/build_wheel.Dockerfile @@ -1,6 +1,6 @@ #syntax=docker/dockerfile:1.1.5-experimental ARG PY_VERSION -FROM tensorflow/build:latest-python$PY_VERSION as base_install +FROM tensorflow/build:2.8-python$PY_VERSION as base_install ENV TF_NEED_CUDA="1" ARG PY_VERSION