Add version constraints for torchvision and lightning

lflautomation · Feb 2, 2024 · 42a4c94 · 42a4c94
1 parent 456696d
commit 42a4c94
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/windows-tests.yml b/.github/workflows/windows-tests.yml
@@ -118,8 +118,11 @@ jobs:
         pip install --progress-bar off .[optional]
         pip install --progress-bar off .[integration]
         pip install "distributed<2023.3.2"
+
         # TODO(nabenabe0928): Remove the version constraint on Torch.
         pip install "torch<2.2.0"
+        pip install "torchvision<0.17.0"
+        pip install "lightning<2.1.4"
 
     - name: Output installed packages
       run: |