diff --git a/.github/workflows/nnAudio.yaml b/.github/workflows/nnAudio.yaml
index d65b99b..6ce7ac2 100644
--- a/.github/workflows/nnAudio.yaml
+++ b/.github/workflows/nnAudio.yaml
@@ -10,7 +10,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.6, 3.7, 3.8, 3.9]
-        torch-version: [1.6.0+cpu]
+        torch-version: [1.6.0]
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}