From 2f34bc44b52d047b430f28727b11d4b5b393ffd3 Mon Sep 17 00:00:00 2001
From: James Thewlis <james@unitary.ai>
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