From 2f34bc44b52d047b430f28727b11d4b5b393ffd3 Mon Sep 17 00:00:00 2001 From: James Thewlis Date: Sat, 8 Jul 2023 11:45:45 +0100 Subject: [PATCH] Update github workflow for PyTorch 2.0.1 --- .github/workflows/test.yml | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 570a54198..8f51da931 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -11,18 +11,20 @@ jobs: runs-on: ubuntu-latest strategy: matrix: - python-version: [3.8] - pytorch-version: [1.7.1, 1.9.1, 1.10.1] + pytorch-version: [1.7.1, 1.9.1, 1.10.1, 2.0.1] include: - - python-version: 3.8 + - python-version: '3.8' pytorch-version: 1.7.1 torchvision-version: 0.8.2 - - python-version: 3.8 + - python-version: '3.8' pytorch-version: 1.9.1 torchvision-version: 0.10.1 - - python-version: 3.8 + - python-version: '3.8' pytorch-version: 1.10.1 torchvision-version: 0.11.2 + - python-version: '3.11' + pytorch-version: 2.0.1 + torchvision-version: 0.15.2 steps: - uses: conda-incubator/setup-miniconda@v2 - run: conda install -n test python=${{ matrix.python-version }} pytorch=${{ matrix.pytorch-version }} torchvision=${{ matrix.torchvision-version }} cpuonly -c pytorch