add key for cuda install

AndrewBoessen · Jan 9, 2025 · 5429ec9 · 5429ec9
1 parent 04cf14e
commit 5429ec9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/makefile.yml b/.github/workflows/makefile.yml
@@ -17,11 +17,11 @@ jobs:
     - name: Install CUDA Toolkit
       run: |
         sudo apt-get update
-        sudo apt-get install -y wget
-        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
-        sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
-        sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
-        sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
+        sudo apt-get install -y wget gnupg2
+        # Add NVIDIA repository key
+        sudo mkdir -p /etc/apt/keyrings
+        wget -qO- https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub | sudo gpg --dearmor -o /etc/apt/keyrings/cuda.gpg
+        echo "deb [signed-by=/etc/apt/keyrings/cuda.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" | sudo tee /etc/apt/sources.list.d/cuda.list
         sudo apt-get update
         sudo apt-get -y install cuda-toolkit-11-7