Update github workflow for PyTorch 2.0.1
This commit is contained in:
parent
a1d071733d
commit
2f34bc44b5
|
|
@ -11,18 +11,20 @@ jobs:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
python-version: [3.8]
|
pytorch-version: [1.7.1, 1.9.1, 1.10.1, 2.0.1]
|
||||||
pytorch-version: [1.7.1, 1.9.1, 1.10.1]
|
|
||||||
include:
|
include:
|
||||||
- python-version: 3.8
|
- python-version: '3.8'
|
||||||
pytorch-version: 1.7.1
|
pytorch-version: 1.7.1
|
||||||
torchvision-version: 0.8.2
|
torchvision-version: 0.8.2
|
||||||
- python-version: 3.8
|
- python-version: '3.8'
|
||||||
pytorch-version: 1.9.1
|
pytorch-version: 1.9.1
|
||||||
torchvision-version: 0.10.1
|
torchvision-version: 0.10.1
|
||||||
- python-version: 3.8
|
- python-version: '3.8'
|
||||||
pytorch-version: 1.10.1
|
pytorch-version: 1.10.1
|
||||||
torchvision-version: 0.11.2
|
torchvision-version: 0.11.2
|
||||||
|
- python-version: '3.11'
|
||||||
|
pytorch-version: 2.0.1
|
||||||
|
torchvision-version: 0.15.2
|
||||||
steps:
|
steps:
|
||||||
- uses: conda-incubator/setup-miniconda@v2
|
- 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
|
- run: conda install -n test python=${{ matrix.python-version }} pytorch=${{ matrix.pytorch-version }} torchvision=${{ matrix.torchvision-version }} cpuonly -c pytorch
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue