Skip to content

Commit 56848a5

Browse files
authored
Changes required torch version (#107)
1 parent 17d50b3 commit 56848a5

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

.github/workflows/tests.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ jobs:
2121
python-version: ${{ matrix.python-version }}
2222
- if: matrix.os == 'windows-latest'
2323
name: Install dependencies - Windows
24-
run: pip install 'torch>=1,<2' -f https://download.pytorch.org/whl/torch_stable.html
24+
run: pip install torch==1.7.0+cpu torchvision==0.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
2525
- name: Install package
2626
run: pip install invoke .[dev]
2727
- name: invoke lint
@@ -58,7 +58,7 @@ jobs:
5858
python-version: ${{ matrix.python-version }}
5959
- if: matrix.os == 'windows-latest'
6060
name: Install dependencies - Windows
61-
run: pip install 'torch>=1,<2' -f https://download.pytorch.org/whl/torch_stable.html
61+
run: pip install torch==1.7.0+cpu torchvision==0.8.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
6262
- name: Install package and dependencies
6363
run: pip install invoke .[test]
6464
- name: invoke pytest

0 commit comments

Comments
 (0)