From 352fb8221f880e686c276449c3c778b3b28b3a6e Mon Sep 17 00:00:00 2001 From: Diego Urgell Date: Tue, 10 Sep 2024 16:26:35 -0700 Subject: [PATCH] attempt fixing unit test version incompatibility (#207) Summary: Pull Request resolved: https://github.com/pytorch/torcheval/pull/207 Differential Revision: D62472378 --- .github/workflows/unit_test.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/unit_test.yaml b/.github/workflows/unit_test.yaml index 38996aad..423b1b42 100644 --- 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]"