diff --git a/.github/workflows/pr-stage-check.yml b/.github/workflows/pr-stage-check.yml index 15669a3f..ec500789 100644 --- a/.github/workflows/pr-stage-check.yml +++ b/.github/workflows/pr-stage-check.yml @@ -20,7 +20,7 @@ jobs: matrix: python-version: ['3.10'] include: - - torch: 2.0.0 + - torch: 2.5.1 steps: - uses: actions/checkout@v3 - name: Set up Python ${{ matrix.python-version }} @@ -30,7 +30,7 @@ jobs: - name: Upgrade pip run: python -m pip install --upgrade pip - name: Install PyTorch - run: pip install torch==${{matrix.torch}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html + run: pip install torch==${{matrix.torch}} -f https://download.pytorch.org/whl/cpu/torch_stable.html - name: Install system dependencies run: | sudo sed -i '$ a deb http://th.archive.ubuntu.com/ubuntu jammy main' /etc/apt/sources.list