Merge branch 'master' of https://github.com/ToddThomson/Mila

ToddThomson · Jan 12, 2025 · 9a79a0b · 9a79a0b
2 parents 33bc854 + 6774da5
commit 9a79a0b
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/.github/workflows/cmake-ubuntu-platform.yml b/.github/workflows/cmake-ubuntu-platform.yml
@@ -9,6 +9,7 @@ on:
 env:
   # Customize the CMake build type here (Release, Debug, RelWithDebInfo, etc.)
   BUILD_TYPE: Release
+  CUDAToolkit_ROOT: /usr/local/cuda-12.6
 
 jobs:
   build:
@@ -22,11 +23,13 @@ jobs:
 
     - name: Install CUDA Toolkit
       run: |
-        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-repo-ubuntu2004_12.6.0-1_amd64.deb
-        sudo dpkg -i cuda-repo-ubuntu2004_12.6.0-1_amd64.deb
-        sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
+        wget https://developer.download.nvidia.com/compute/cuda/12.6.3/local_installers/cuda-repo-ubuntu2404-12-6-local_12.6.3-560.35.05-1_amd64.deb
+        sudo dpkg -i cuda-repo-ubuntu2404-12-6-local_12.6.3-560.35.05-1_amd64.deb
+        sudo cp /var/cuda-repo-ubuntu2404-12-6-local/cuda-*-keyring.gpg /usr/share/keyrings/
         sudo apt-get update
-        sudo apt-get -y install cuda-12-6
+        sudo apt-get -y install cuda-toolkit-12-6
+        echo "CUDAToolkit_ROOT=/usr/local/cuda-12.6" >> $GITHUB_ENV
+        echo "PATH=/usr/local/cuda-12.6/bin:$PATH" >> $GITHUB_ENV
 
     - name: Configure CMake
       run: cmake -B ${{github.workspace}}/build -DCMAKE_BUILD_TYPE=${{env.BUILD_TYPE}}
@@ -36,4 +39,4 @@ jobs:
 
     - name: Test
       working-directory: ${{github.workspace}}/build
-      run: ctest -C ${{env.BUILD_TYPE}}
+      run: ctest -C ${{env.BUILD_TYPE}}