From c1c10c89c815156dc736224cdff21a2a426ab2eb Mon Sep 17 00:00:00 2001 From: Sebastian Raschka Date: Thu, 30 Jan 2025 12:45:48 -0600 Subject: [PATCH] Test PyTorch nightly releases (#509) --- .github/workflows/basic-tests-pytorch-rc.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/basic-tests-pytorch-rc.yml b/.github/workflows/basic-tests-pytorch-rc.yml index f99c733..bc211f1 100644 --- a/.github/workflows/basic-tests-pytorch-rc.yml +++ b/.github/workflows/basic-tests-pytorch-rc.yml @@ -1,4 +1,4 @@ -name: Test latest PyTorch release candidate +name: Test latest PyTorch nightly / release candidate on: push: branches: [ main ] @@ -35,7 +35,7 @@ jobs: pip install pytest nbval if [ -f requirements.txt ]; then pip install -r requirements.txt; fi pip install -r ch05/07_gpt_to_llama/tests/test-requirements-extra.txt - pip install torch==2.6.0 torchvision --index-url https://download.pytorch.org/whl/test/cpu + pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu - name: Test Selected Python Scripts run: |