diff --git a/.github/workflows/build-and-test.yaml b/.github/workflows/build-and-test.yaml
index 2c14f61..31dc9ec 100644
--- a/.github/workflows/build-and-test.yaml
+++ b/.github/workflows/build-and-test.yaml
@@ -25,10 +25,6 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install -r ./requirements.txt
-    - uses: Jimver/cuda-toolkit@v0.2.10
-      id: cuda-toolkit
-      with:
-        cuda: '12.1.0'
     - name: Run tests
       run: |
         pytest --cov-config=.coveragerc --cov-report=xml --cov=abl ./tests 
diff --git a/tests/test_models.py b/tests/test_models.py
index b01b8c3..8ea4383 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -18,7 +18,7 @@ class TestBasicModel(object):
     @pytest.mark.parametrize("cls", [LeNet5, SymbolNet])
     @pytest.mark.parametrize("criterion", [nn.CrossEntropyLoss])
     @pytest.mark.parametrize("optimizer", [torch.optim.RMSprop])
-    @pytest.mark.parametrize("device", [torch.device("cpu"), torch.device("cuda:0")])
+    @pytest.mark.parametrize("device", [torch.device("cpu")])
     def test_models(self, num_classes, image_size, cls, criterion, optimizer, device):
         cls = cls(num_classes=num_classes, image_size=image_size)
         criterion = criterion()