install python via conda

rapidsai · Nov 6, 2023 · 1335b56 · 1335b56
1 parent 510db4a
commit 1335b56
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
@@ -29,7 +29,7 @@ jobs:
         python-version: ["3.10", "3.11", "3.12"]
     container:
       # Use CentOS 7 for an old glibc version, so the wheel has broad manylinux compatibility.
-      image: "rapidsai/ci-conda:cuda12.0.1-centos7-py${{ matrix.python-version }}"
+      image: "rapidsai/ci-conda:cuda12.0.1-centos7-py3.10
     steps:
     - name: Checkout code
       uses: actions/checkout@v3
@@ -39,4 +39,4 @@ jobs:
       id: get-pr-info
       uses: rapidsai/shared-actions/get-pr-info@main
     - name: Run build_wheel.sh
-      run: ci/build_wheel.sh
+      run: ci/build_wheel.sh ${{ matrix.python-version }}
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -3,6 +3,9 @@
 
 set -e
 
+python_version="${1:-3.10}"
+conda install -y python="$python_version"
+
 rapids-logger "Install CUDA Toolkit"
 yum update
 yum install -y epel-release