TORCH_VERSION=2.2.1

Author: djherbis

tpu/config.txt

@@ -7,7 +7,7 @@ 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.2.0
+TORCH_VERSION=2.2.1
 # https://github.com/pytorch/audio supports nightly
 TORCHAUDIO_VERSION=2.2.1
 # https://github.com/pytorch/text supports main