Skip to content

Commit 17a434b

Browse files
update databricks init script
Signed-off-by: Jaya Venkatesh <jjayabaskar@nvidia.com>
1 parent 47c1529 commit 17a434b

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

source/conf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
"rapids_conda_channels": "-c rapidsai-nightly -c conda-forge",
5050
"rapids_conda_packages": f"rapids={nightly_version} python=3.13 'cuda-version>=12.0,<=12.9'",
5151
"rapids_pip_index": "https://pypi.anaconda.org/rapidsai-wheels-nightly/simple",
52-
"rapids_pip_version": f"~={nightly_version}.0a0",
52+
"rapids_pip_version": f"{nightly_version}.*,>=0.0.0a0",
5353
# SageMaker Notebook Instance examples need to stay pinned to an older RAPIDS until this is resolved:
5454
# https://github.com/rapidsai/deployment/issues/520
5555
"rapids_sagemaker_conda_packages": f"rapids={nightly_version} python=3.12 cuda-version=13",

source/platforms/databricks.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ set -e
2929
# Install RAPIDS libraries
3030
pip install \
3131
--extra-index-url={{rapids_pip_index}} \
32-
"cudf-cu12>={{rapids_pip_version}}" "cuml-cu12>={{rapids_pip_version}}" \
33-
"dask-cuda>={{rapids_pip_version}}"
32+
"cudf-cu12=={{rapids_pip_version}}" "cuml-cu12=={{rapids_pip_version}}" \
33+
"dask-cuda=={{rapids_pip_version}}"
3434

3535
```
3636

0 commit comments

Comments
 (0)