Add py3.10 to GPU image

lan200485 · Apr 3, 2023 · 4a71401 · 4a71401
1 parent 9552e2c
commit 4a71401
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config.txt b/config.txt
@@ -1,7 +1,7 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
 BASE_IMAGE_TAG=m105
 CPU_BASE_IMAGE_NAME=tf2-cpu.2-11.py310
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-11
+GPU_BASE_IMAGE_NAME=tf2-gpu.2-11.py310
 LIGHTGBM_VERSION=3.3.2
 TORCH_VERSION=1.13.0
 TORCHAUDIO_VERSION=0.13.0