Update CUDA to minimum version present on image for Ubuntu 22.04

raulcd · Sep 16, 2024 · 4f1cd9c · 4f1cd9c
1 parent 336b9bb
commit 4f1cd9c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/dev/tasks/python-sdist/github.yml b/dev/tasks/python-sdist/github.yml
@@ -22,7 +22,7 @@
 jobs:
   build:
     name: "Build sdist"
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
     steps:
       {{ macros.github_checkout_arrow()|indent }}
       {{ macros.github_install_archery()|indent }}
@@ -39,6 +39,7 @@ jobs:
       - name: Test sdist
         run: archery docker run ubuntu-python-sdist-test
         env:
+          UBUNTU: 22.04
           PYARROW_VERSION: {{ arrow.no_rc_version }}
 
       {{ macros.github_upload_releases("arrow/python/dist/*.tar.gz")|indent }}

diff --git a/dev/tasks/tasks.yml b/dev/tasks/tasks.yml
@@ -1514,20 +1514,28 @@ tasks:
 
   ############################## CUDA tests #################################
 
-  test-cuda-cpp:
+  test-cuda-cpp-ubuntu-20.04-cuda-11.2.2:
+    ci: github
+    template: docker-tests/github.cuda.yml
+    params:
+      image: ubuntu-cuda-cpp
+
+  test-cuda-cpp-ubuntu-22.04-cuda-11.7.1:
     ci: github
     template: docker-tests/github.cuda.yml
     params:
       env:
         UBUNTU: 22.04
+        CUDA: 11.7.1
       image: ubuntu-cuda-cpp
 
-  test-cuda-python:
+  test-cuda-python-ubuntu-22.04-cuda-11.7.1:
     ci: github
     template: docker-tests/github.cuda.yml
     params:
       env:
         UBUNTU: 22.04
+        CUDA: 11.7.1
       image: ubuntu-cuda-python
 
   ############################## Fuzz tests #################################