From b01bcb52fffa67920e232ca9424d3796ef302537 Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Tue, 15 Feb 2022 09:36:47 -0800 Subject: [PATCH] I love consistent build containers --- 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 cf186fddff..c4eca2eb88 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:2.8-python$PY_VERSION as base_install +FROM tensorflow/build:latest-python$PY_VERSION as base_install ENV TF_NEED_CUDA="1" ARG PY_VERSION