diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
index a09f770..4ad4f0e 100644
--- 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: |