This commit is contained in:
James Thewlis 2024-06-11 15:40:37 +03:00 committed by GitHub
commit d05473ef7a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 5 deletions

View File

@ -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