diff --git a/.github/workflows/linux_gpu.yaml b/.github/workflows/linux_gpu.yaml
index d31a878..2eb0e55 100644
--- a/.github/workflows/linux_gpu.yaml
+++ b/.github/workflows/linux_gpu.yaml
@@ -39,10 +39,10 @@ jobs:
           wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
           sudo dpkg -i cuda-keyring_1.1-1_all.deb
           sudo apt-get update
-          sudo apt-get -y install cuda-toolkit-12-4 nvidia-open
-
-      - name: Check CUDA info
-        run: nvidia-smi
+          sudo apt-get -y install nvidia-open
+          sudo reboot
+          sudo apt-get -y install cuda-toolkit-12-4
+          nvidia-smi
 
       - name: Download ISPC
         run: |