Update pytest.yml

KennardWang · Oct 22, 2023 · 911fa53 · 911fa53
1 parent 4e4370e
commit 911fa53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -32,7 +32,7 @@ jobs:
           pip install -r requirements.txt
           conda install -c conda-forge dlib
           pip install scipy>0.16
-          pip install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102
+          conda install pytorch==1.10.2 torchvision==0.11.3 torchaudio==0.10.2 cudatoolkit=10.2 -c pytorch
     
       - name: Run tests
         run: |
@@ -61,7 +61,7 @@ jobs:
           pip install -r requirements.txt
           conda install -c conda-forge dlib
           pip install scipy>0.16
-          pip install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102
+          conda install pytorch==1.10.2 torchvision==0.11.3 torchaudio==0.10.2 -c pytorch
     
       - name: Run tests
         run: |