Skip to content

[TPU] move torch_xla out of tpu.txt #26938

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
yaochengji wants to merge 3 commits into from
Closed

[TPU] move torch_xla out of tpu.txt #26938

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
41fc90c
[TPU] move torch_xla out of tpu.txt
yaochengji Oct 15, 2025
4d8e9a2
fix dockerfile
yaochengji Oct 16, 2025
cb8529e
fix dockerfile
yaochengji Oct 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/getting_started/installation/google_tpu.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,7 @@ Install build dependencies:

```bash
pip install -r requirements/tpu.txt
pip install torch_xla[tpu, pallas]==2.8.0
sudo apt-get install --no-install-recommends --yes libopenblas-base libopenmpi-dev libomp-dev
```

Expand Down
3 changes: 0 additions & 3 deletions requirements/tpu.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,3 @@ ray[data]
setuptools==78.1.0
nixl==0.3.0
tpu_info==0.4.0
Comment on lines 12 to 14
Copy link

@chatgpt-codex-connector chatgpt-codex-connector bot Oct 15, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P1 Badge Keep torch_xla installation in TPU image build

Removing torch_xla[tpu, pallas]==2.8.0 from the TPU requirements means the Docker build that uses this file (docker/Dockerfile.tpu) will now uninstall torch_xla and never reinstall it. The Dockerfile still runs pip uninstall -y torch torch_xla torchvision before pip install -r requirements/tpu.txt, so the resulting image no longer contains torch_xla and any TPU runtime code that imports it will fail at startup. Consider either leaving torch_xla in this requirements file or explicitly installing it again inside the Dockerfile.

Useful? React with 👍 / 👎.


# Install torch_xla
torch_xla[tpu, pallas]==2.8.0
Loading