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

2025-08-18 16:23:33 -07:00
parent 6d25e3fd6e
commit 498259ccce
1 changed files with 1 additions and 0 deletions
--- 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