From 6115751bf92dbba858f3121c541a7903e123b2aa Mon Sep 17 00:00:00 2001 From: bhack Date: Sat, 11 May 2024 13:12:06 +0200 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 9542e1365a..ef64befbd2 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.15-python$PY_VERSION as base_install +FROM tensorflow/build:2.16-python$PY_VERSION as base_install ENV TF_NEED_CUDA="1" ARG PY_VERSION