From 15e057c79e45016bdb4200a773b3dbf9f3e1dbef Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Mon, 26 Feb 2024 11:49:33 -0800 Subject: [PATCH] Bump PyTorch version to 2.2 --- tpu/config.txt | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tpu/config.txt b/tpu/config.txt index a544de0f..6d6bc722 100644 --- a/tpu/config.txt +++ b/tpu/config.txt @@ -7,11 +7,11 @@ TF_LIBTPU_VERSION=1.9.0 TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64 JAX_VERSION=0.4.23 # Supports nightly -TORCH_VERSION=2.1.0 +TORCH_VERSION=2.2.0 # https://github.com/pytorch/audio supports nightly -TORCHAUDIO_VERSION=2.1.0 +TORCHAUDIO_VERSION=2.2.1 # https://github.com/pytorch/text supports main -TORCHTEXT_VERSION=0.16.0 +TORCHTEXT_VERSION=0.17.1 # https://github.com/pytorch/vision supports nightly -TORCHVISION_VERSION=0.16.0 +TORCHVISION_VERSION=0.17.1 TORCH_LINUX_WHEEL_VERSION=manylinux_2_28_x86_64