diff --git a/config.txt b/config.txt index 1a9474ab..b6b788f2 100644 --- a/config.txt +++ b/config.txt @@ -2,7 +2,7 @@ BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release BASE_IMAGE_TAG=m86 CPU_BASE_IMAGE_NAME=tf2-cpu.2-6 GPU_BASE_IMAGE_NAME=tf2-gpu.2-6 -LIGHTGBM_VERSION=3.2.1 +LIGHTGBM_VERSION=3.3.1 TORCH_VERSION=1.9.1 TORCHAUDIO_VERSION=0.9.1 TORCHTEXT_VERSION=0.10.1