This is an automated email from the ASF dual-hosted git repository.
aglinxinyuan pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/texera.git
The following commit(s) were added to refs/heads/main by this push:
new ea110e798a chore(dependencies): pin torch to CPU wheel on Linux x86_64
(#4473)
ea110e798a is described below
commit ea110e798ab4b5908641875f4ef607550dc36d5a
Author: Jiadong Bai <[email protected]>
AuthorDate: Wed Apr 22 17:04:43 2026 -0700
chore(dependencies): pin torch to CPU wheel on Linux x86_64 (#4473)
### What changes were proposed in this PR?
Pin torch to PyTorch's CPU wheel on Linux x86_64 so the proprietary
NVIDIA CUDA libraries (`nvidia-*-cu12`) are not installed.
Other platforms fall through to plain `torch==2.8.0` from PyPI, which is
already a CPU build.
### Any related issues, documentation, discussions?
Closes #4472.
### How was this PR tested?
`pip install --dry-run -r amber/operator-requirements.txt` installs no
`nvidia-*` or `triton` packages on either macOS (picks plain 2.8.0) or
Linux x86_64 (picks 2.8.0+cpu).
### Was this PR authored or co-authored using generative AI tooling?
Generated-by: Claude Code (Claude Opus 4.7)
---------
Co-authored-by: Claude Opus 4.7 (1M context) <[email protected]>
Co-authored-by: Xinyuan Lin <[email protected]>
---
amber/operator-requirements.txt | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/amber/operator-requirements.txt b/amber/operator-requirements.txt
index 2554373e0e..a729639466 100644
--- a/amber/operator-requirements.txt
+++ b/amber/operator-requirements.txt
@@ -20,7 +20,12 @@ plotly==5.24.1
praw==7.6.1
pillow==12.1.1
pybase64==1.3.2
-torch==2.8.0
+
+# Pin torch to the CPU wheel on Linux x86_64 to avoid the NVIDIA CUDA deps.
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch==2.8.0+cpu ; platform_system == "Linux" and platform_machine == "x86_64"
+torch==2.8.0 ; platform_system != "Linux" or platform_machine != "x86_64"
+
scikit-learn==1.5.0
transformers==4.57.3
boto3==1.40.53
