Unpin torchvision (#2077)

ludwig-ai · May 30, 2022 · 8e4bf24 · 8e4bf24
1 parent a36fbb2
commit 8e4bf24
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -87,9 +87,7 @@ jobs:
           if [ $PYTORCH == "nightly" ]; then
             cat requirements.txt | sed '/^torch[>=<]/d' > requirements-temp && mv requirements-temp requirements.txt
             extra_index_url=https://download.pytorch.org/whl/nightly/cpu
-            pip install --pre torch==1.13.0.dev20220527+cpu torchtext torchaudio --extra-index-url $extra_index_url
-            # TODO: temp fix for https://github.com/pytorch/vision/issues/6103
-            pip install https://download.pytorch.org/whl/nightly/cpu/torchvision-0.14.0.dev20220527%2Bcpu-cp39-cp39-linux_x86_64.whl
+            pip install --pre torch torchtext torchvision torchaudio --extra-index-url $extra_index_url
           else
             extra_index_url=https://download.pytorch.org/whl/cpu
             pip install torch==$PYTORCH torchtext torchvision torchaudio --extra-index-url $extra_index_url