diff --git a/.github/workflows/tests-nightly.yml b/.github/workflows/tests-nightly.yml index 00b624243..fc6c2b396 100644 --- a/.github/workflows/tests-nightly.yml +++ b/.github/workflows/tests-nightly.yml @@ -52,7 +52,7 @@ jobs: - name: Install PyTorch run: | - pip3 install torch==${{ matrix.pytorch_version }} + pip3 install torch==${{ matrix.pytorch_version }} torchvision - name: Update requirements.txt run: | diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 9c983ad70..d4818f280 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -78,7 +78,7 @@ jobs: - name: Install PyTorch run: | - pip3 install torch==${{ matrix.pytorch_version }} + pip3 install torch==${{ matrix.pytorch_version }} torchvision - name: Install dependencies run: | @@ -151,7 +151,7 @@ jobs: - name: Install PyTorch run: | - pip3 install torch==${{ matrix.pytorch_version }} + pip3 install torch==${{ matrix.pytorch_version }} torchvision - name: Install dependencies run: |