Install tpu_info==0.4.0 to fix core dump for TPU (#23135)

xiangxu-google · web-flow · commit 498259ccce7f · 2025-08-18T16:23:33.000-07:00
diff --git a/requirements/tpu.txt b/requirements/tpu.txt
@@ -11,6 +11,7 @@ ray[default]
 ray[data]
 setuptools==78.1.0
 nixl==0.3.0
+tpu_info==0.4.0
 
 # Install torch_xla
 --pre
