diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 55db40f..759cf98 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -25,7 +25,8 @@ jobs:
         wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
         sudo dpkg -i cuda-keyring_1.0-1_all.deb
         sudo apt-get update
-        # Install the CUDA Toolkit 11.0
+        # Install the NVIDIA and CUDA Toolkit 11.0
+        sudo apt-get install -y nvidia-driver-470
         sudo apt-get install -y cuda-toolkit-11-0  # Install CUDA 11.0
         echo "/usr/local/cuda/bin" >> $GITHUB_PATH
         echo "/usr/local/cuda/lib64" >> $GITHUB_PATH