diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 9af767db..a5cc070b 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -32,25 +32,12 @@ jobs:
           
           sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
           sudo update-alternatives --set c++ /usr/bin/g++
-                    
-          # Install CUDA
-          
-          wget --no-verbose https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
-          bash cuda_11.8.0_520.61.05_linux.run
-          rm cuda_11.8.0_520.61.05_linux.run
-          
-          # Install cudnn
-          tar -xf cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz
-          mv cudnn-linux-x86_64-8.9.2.26_cuda11-archive cudnn
-          cp cudnn/include/cudnn*.h /usr/local/cuda/include
-          cp cudnn/lib/libcudnn* /usr/local/cuda/lib64
-          chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
-          rm -rf cudnn*
-          
-          # Copy these to ~/.profile
-          echo "LD_LIBRARY_PATH=$(echo $LD_LIBRARY_PATH:/usr/local/cuda/lib64)" >> $GITHUB_ENV
-          echo "CUDA_HOME=/usr/local/cuda" >> $GITHUB_ENV 
-          echo "PATH=$(echo /usr/local/cuda/bin:$PATH) >> $GITHUB_ENV
+
+      - name: Install cuda
+        uses: Jimver/cuda-toolkit@v0.2.10
+        id: cuda-toolkit
+        with:
+          cuda: '11.8.0'
 
       - name: Set up Bazel
         uses: bazelbuild/setup-bazelisk@v2