Force cuda-version.

jameslamb · Jan 30, 2024 · d4f8aa2 · d4f8aa2
1 parent 1050767
commit d4f8aa2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/ci/test_cpp.sh b/ci/test_cpp.sh
@@ -26,6 +26,7 @@ rapids-print-env
 
 rapids-mamba-retry install \
   --channel "${CPP_CHANNEL}" \
+  cuda-version=${RAPIDS_CUDA_VERSION%.*} \
   librmm librmm-tests
 
 rapids-logger "Check GPU usage"

diff --git a/ci/test_python.sh b/ci/test_python.sh
@@ -24,6 +24,7 @@ PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
 rapids-mamba-retry install \
   --channel "${CPP_CHANNEL}" \
   --channel "${PYTHON_CHANNEL}" \
+  cuda-version=${RAPIDS_CUDA_VERSION%.*} \
   rmm librmm
 
 RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}