Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions ci/stable_install/install_and_test_conda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ set -euo pipefail

SCRIPT_DIR="$(dirname "${BASH_SOURCE[0]}")"
source "${SCRIPT_DIR}/test_imports.sh"
source "${SCRIPT_DIR}/install_rapids_doctor.sh"

STABLE_RAPIDS_VERSION="26.4.*"

Expand Down Expand Up @@ -79,5 +80,11 @@ declare -a RAPIDS_IMPORTS=(
)
testImports RAPIDS_IMPORTS

# Run RAPIDS health checks
installRapidsDoctor

rapids-logger "Running RAPIDS doctor smoke tests"
rapids doctor --verbose

conda deactivate
conda env remove -n "$envName"
11 changes: 9 additions & 2 deletions ci/stable_install/install_and_test_pip.sh
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ STABLE_RAPIDS_VERSION="26.4.*"
SCRIPT_DIR="$(dirname "${BASH_SOURCE[0]}")"
source "${SCRIPT_DIR}/bootstrap/pip.sh"
source "${SCRIPT_DIR}/test_imports.sh"
source "${SCRIPT_DIR}/install_rapids_doctor.sh"

while [[ $# -gt 0 ]]; do
case $1 in
Expand Down Expand Up @@ -91,7 +92,7 @@ createPyEnv "$PYTHON_VERSION" "$INSTALL_DIR"
rapids-logger "Downloading NVIDIA PyPI only wheels for Python $PYTHON_VERSION and CUDA $CUDA_VERSION"

WHEELS_DIR=$(mktemp -d)
pip download \
python -m pip download \
--isolated \
--index-url https://pypi.nvidia.com \
--prefer-binary \
Expand All @@ -108,7 +109,7 @@ pip download \

rapids-logger "Testing stable version install with Python $PYTHON_VERSION and CUDA $CUDA_VERSION"

pip install \
python -m pip install \
--isolated \
--index-url https://pypi.org/simple \
--prefer-binary \
Expand All @@ -133,6 +134,12 @@ declare -a RAPIDS_IMPORTS=(
)
testImports RAPIDS_IMPORTS

# Run RAPIDS health checks
installRapidsDoctor

rapids-logger "Running RAPIDS doctor smoke tests"
rapids doctor --verbose

popd

rapids-logger "Removing environment in $INSTALL_DIR/.venv"
Expand Down
10 changes: 10 additions & 0 deletions ci/stable_install/install_rapids_doctor.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash
# Copyright (c) 2026, NVIDIA CORPORATION.

function installRapidsDoctor {
# Clone and install `rapids-cli`
git clone https://github.com/rapidsai/rapids-cli rapids-cli
pushd rapids-cli || exit 1
python -m pip install .
popd || exit 1
}
Loading