diff --git a/.github/workflows/unittest.yml b/.github/workflows/unittest.yml
index a139565..0ddbc1a 100644
--- a/.github/workflows/unittest.yml
+++ b/.github/workflows/unittest.yml
@@ -31,7 +31,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install torch==1.8.1 osqp==0.6.2.post8 pygame==2.3.0 numpy==1.16.6 scipy==1.1.0
+        pip install torch==1.8.1+cpu osqp==0.6.2.post8 pygame==2.3.0 numpy==1.16.6 scipy==1.1.0
 
     # Run unittests using the Python unittest module
     - name: Run unittest