Fix CUDA install in CI

juntyr · May 18, 2024 · f395253 · f395253
1 parent 90717eb
commit f395253
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 20 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -98,14 +98,11 @@ jobs:
 
     steps:
       - name: Install CUDA
-        run: |
-          wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
-          sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
-          curl -L -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb
-          sudo dpkg -i cuda-keyring_1.0-1_all.deb
-          sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
-          sudo apt-get update -q
-          sudo apt-get install cuda -y --no-install-recommends
+        uses: Jimver/cuda-toolkit@v0.2.14
+        with:
+          method: network
+          use-github-cache: false
+          use-local-cache: false
 
       - name: Checkout the Repository
         uses: actions/checkout@v2

diff --git a/.github/workflows/coverage.yml b/.github/workflows/coverage.yml
@@ -11,14 +11,11 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Install CUDA
-        run: |
-          wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
-          sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
-          curl -L -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb
-          sudo dpkg -i cuda-keyring_1.0-1_all.deb
-          sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
-          sudo apt-get update -q
-          sudo apt-get install cuda -y --no-install-recommends
+        uses: Jimver/cuda-toolkit@v0.2.14
+        with:
+          method: network
+          use-github-cache: false
+          use-local-cache: false
 
       - name: Checkout the Repository
         uses: actions/checkout@v2

diff --git a/src/lib.rs b/src/lib.rs
@@ -3,10 +3,7 @@
 #![cfg_attr(not(feature = "host"), no_std)]
 #![feature(auto_traits)]
 #![feature(negative_impls)]
-#![cfg_attr(
-    any(all(not(feature = "host"), target_os = "cuda"), doc),
-    feature(stdarch_nvptx)
-)]
+#![cfg_attr(all(not(feature = "host"), target_os = "cuda"), feature(stdarch_nvptx))]
 #![cfg_attr(any(feature = "alloc", doc), feature(allocator_api))]
 #![feature(doc_cfg)]
 #![feature(marker_trait_attr)]