From a569dabdc81a84c75ff8d9e33b72ab3a89e97680 Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Fri, 20 May 2022 16:08:56 -0700 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 73d228d321..1c2d28818d 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.9-python$PY_VERSION as base_install ENV TF_NEED_CUDA="1" ARG PY_VERSION