Update release.yml

soedinglab · Nov 29, 2024 · 4b39ff7 · 4b39ff7
1 parent 80bf735
commit 4b39ff7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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