From e4f3647f9a01072d6a3fe17d58b578964f5d120e Mon Sep 17 00:00:00 2001 From: Jamie Pine Date: Mon, 2 Feb 2026 00:17:57 -0800 Subject: [PATCH] Add CPU-only PyTorch installation for Linux in release workflow - Included a conditional installation of CPU-only PyTorch packages for Ubuntu 22.04 to reduce unnecessary CUDA dependencies. - Updated the release workflow to ensure compatibility with CPU-focused builds on Linux. --- .github/workflows/release.yml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 177e684d..4ee9475f 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -64,6 +64,10 @@ jobs: run: | python -m pip install --upgrade pip pip install pyinstaller + # Install CPU-only torch on Linux to avoid CUDA bloat + if [ "${{ matrix.platform }}" == "ubuntu-22.04" ]; then + pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu + fi pip install -r providers/pytorch-cpu/requirements.txt pip install -r backend/requirements.txt