attempt fixing unit test version incompatibility (pytorch#207)

Summary: Pull Request resolved: pytorch#207 Differential Revision: D62472378
diego-urgell · Sep 10, 2024 · 352fb82 · 352fb82
1 parent a88a579
commit 352fb82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/unit_test.yaml b/.github/workflows/unit_test.yaml
@@ -25,7 +25,7 @@ jobs:
         run: |
           set -eux
           conda activate test
-          conda install pytorch torchaudio torchvision cpuonly -c pytorch-nightly
+          conda install pytorch torchaudio torchvision cpuonly -c pytorch-nightly -c nvidia
           pip install -r requirements.txt
           pip install -r dev-requirements.txt
           pip install --no-build-isolation -e ".[dev]"