Skip to content

chore(dependencies): pin torch to CPU wheel on Linux x86_64#4473

Merged
aglinxinyuan merged 3 commits into
apache:mainfrom
bobbai00:chore/torch-cpu-only
Apr 23, 2026
Merged

chore(dependencies): pin torch to CPU wheel on Linux x86_64#4473
aglinxinyuan merged 3 commits into
apache:mainfrom
bobbai00:chore/torch-cpu-only

Conversation

@bobbai00

@bobbai00 bobbai00 commented Apr 22, 2026

Copy link
Copy Markdown
Contributor

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)

PyPI's torch 2.8.0 wheel for Linux x86_64 transitively installs the
proprietary NVIDIA CUDA shared libraries (nvidia-*-cu12), which are
not ASF-compatible for binary redistribution. Pin to the CPU-only
wheel from PyTorch's CPU index on that platform via a PEP 508 marker;
other platforms continue to resolve torch 2.8.0 from PyPI, which is
already a CPU build on macOS/Windows/Linux-aarch64.

GPU users install the CUDA wheel themselves post-install.

Closes apache#4472.

Generated-by: Claude Code (Claude Opus 4.7)

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
@bobbai00 bobbai00 requested a review from aglinxinyuan April 22, 2026 23:53
@bobbai00 bobbai00 self-assigned this Apr 22, 2026
@bobbai00 bobbai00 added dependencies Pull requests that update a dependency file pyamber labels Apr 22, 2026
@github-actions github-actions Bot removed dependencies Pull requests that update a dependency file pyamber labels Apr 22, 2026
Comment thread amber/operator-requirements.txt

@aglinxinyuan aglinxinyuan left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM!

@bobbai00 bobbai00 requested a review from aglinxinyuan April 23, 2026 00:01
@aglinxinyuan aglinxinyuan enabled auto-merge (squash) April 23, 2026 00:02
@aglinxinyuan aglinxinyuan merged commit ea110e7 into apache:main Apr 23, 2026
11 checks passed
bobbai00 added a commit to bobbai00/texera that referenced this pull request Apr 23, 2026
…nt deps

Removes entries that are no longer bundled after the following upstream PRs:
- apache#4463 drop `twittered` (and its Sonar/jtidy/Maven-tooling transitive chain)
- apache#4474 drop `python-lsp-server` (and its pylint/rope/astroid chain + extras)
- apache#4475 drop `jschardet` from the frontend
- apache#4473 pin torch to CPU wheel on Linux x86_64 (removes nvidia-*-cu12)
- apache#4386 drop `mysql-connector-java`

Adds entries for newly-bundled deps surfaced by the reconciliation:
- `org.eclipse.collections` + `-api` under EDL-1.0
- Python `imageio` under BSD-2-Clause
- Python `lazy-loader`, `scikit-image`, `tifffile` under BSD-3-Clause

Also drops the now-redundant NOTICE-binary W3C block for xml-apis, the
entire NVIDIA CUDA section in LICENSE-binary, and the Java HTML Tidy
License section.

Verified: the checker reports zero STALE/ADDED across JVM, Python, and
npm ecosystems against a fresh `sbt dist` / frontend build / pip install.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
bobbai00 added a commit to bobbai00/texera that referenced this pull request Apr 23, 2026
…nt deps

Removes entries that are no longer bundled after the following upstream PRs:
- apache#4463 drop `twittered` (and its Sonar/jtidy/Maven-tooling transitive chain)
- apache#4474 drop `python-lsp-server` (and its pylint/rope/astroid chain + extras)
- apache#4475 drop `jschardet` from the frontend
- apache#4473 pin torch to CPU wheel on Linux x86_64 (removes nvidia-*-cu12)
- apache#4386 drop `mysql-connector-java`

Adds entries for newly-bundled deps surfaced by the reconciliation:
- `org.eclipse.collections` + `-api` under EDL-1.0
- Python `imageio` under BSD-2-Clause
- Python `lazy-loader`, `scikit-image`, `tifffile` under BSD-3-Clause

Also drops the now-redundant NOTICE-binary W3C block for xml-apis, the
entire NVIDIA CUDA section in LICENSE-binary, and the Java HTML Tidy
License section.

Verified: the checker reports zero STALE/ADDED across JVM, Python, and
npm ecosystems against a fresh `sbt dist` / frontend build / pip install.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
yangzhang75 pushed a commit to yangzhang75/texera that referenced this pull request Jun 22, 2026
)

### 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 apache#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) <noreply@anthropic.com>
Co-authored-by: Xinyuan Lin <xinyual3@uci.edu>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

NVIDIA CUDA shared libraries bundled via torch on Linux x86_64

2 participants