f

argumentcomputer · Dec 4, 2024 · d4e6880 · d4e6880
1 parent 7128135
commit d4e6880
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -11,11 +11,12 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - run: lsb_release -a
-      - name: Test GPU
-        run: nvidia-smi
+      - run: df -h
+      - run: nvidia-smi
+      - run: sudo apt-get update && sudo apt-get upgrade
       - run: |
-          sudo apt-get update && sudo apt-get upgrade
-      - run: |
-          wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
-          sudo sh cuda_12.4.0_550.54.14_linux.run
+          wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
+          sudo dpkg -i cuda-keyring_1.1-1_all.deb
+          sudo apt-get update
+          sudo apt-get -y install cuda-toolkit-12-4
       - run: which nvcc