Update build_wheel.Dockerfile

tensorflow · Feb 15, 2022 · bd6f2b2 · bd6f2b2
1 parent 4b03e26
commit bd6f2b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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