Build python using conda in GPU jobs

Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
rapidsai · Apr 5, 2022 · 229c67e · 229c67e
1 parent 921d286
commit 229c67e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
@@ -111,12 +111,12 @@ else
         fi
     done
 
-    cd $WORKSPACE/python
-    export LIBRMM_BUILD_DIR="$WORKSPACE/ci/artifacts/rmm/cpu/conda_work/build"
-
-    gpuci_logger "Building rmm"
-    "$WORKSPACE/build.sh" -v rmm
+    gpuci_logger "Building and installing rmm"
+    gpuci_conda_retry build --no-build-id --croot ${CONDA_BLD_DIR} \
+      -c $WORKSPACE/ci/artifacts/rmm/cpu/.conda-bld/ conda/recipes/rmm
+    gpuci_mamba_retry install -c $WORKSPACE/ci/artifacts/rmm/cpu/.conda-bld/ rmm
 
+    cd $WORKSPACE/python
     gpuci_logger "pytest rmm"
     py.test --cache-clear --junitxml=${WORKSPACE}/test-results/junit-rmm.xml -v --cov-config=.coveragerc --cov=rmm --cov-report=xml:${WORKSPACE}/python/rmm-coverage.xml --cov-report term
     exitcode=$?