Update action and CUDA versions

Author: camille-004

.github/workflows/ci.yml

@@ -21,10 +21,10 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v6
 
       - name: Set up Python
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: "3.10"
 
@@ -50,29 +50,27 @@ jobs:
   build-wheel:
     name: Build wheel (Linux x86_64, Python ${{ matrix.python-version }})
     runs-on: ubuntu-latest
+    container:
+      image: nvidia/cuda:13.0.2-base-ubuntu24.04
     strategy:
       fail-fast: false
       matrix:
         python-version: ["3.10"]
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v6
 
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v5
-        with:
-          python-version: ${{ matrix.python-version }}
-
-      - name: Install build dependencies
+      - name: Install Python
         run: |
-          python -m pip install --upgrade pip
-          pip install build wheel setuptools
+          apt-get update
+          apt-get install -y python${{ matrix.python-version }} python${{ matrix.python-version }}-venv python3-pip
+          python${{ matrix.python-version }} -m pip install --upgrade pip build wheel setuptools
 
       - name: Build wheel
-        run: python -m build --wheel --outdir dist/
+        run: python${{ matrix.python-version }} -m build --wheel --outdir dist/
 
       - name: Upload wheel artifact
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v5
         with:
           name: wheel-linux-x86_64-py${{ matrix.python-version }}
           path: dist/*.whl