🔨🐛Pinned pytorch in test

carefree0910 · Oct 17, 2024 · ada6474 · ada6474
1 parent 080b451
commit ada6474
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -24,7 +24,7 @@ jobs:
       - name: Install dependencies
         shell: bash -l {0}
         run: |
-          conda install pytorch torchvision torchaudio cpuonly -c pytorch
+          conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 -c pytorch
           pip install -e .
           pip install -U --force-reinstall chardet charset-normalizer
       - name: Install test dependencies