Tweak Databricks install instructions (#482)

Co-authored-by: James Lamb <[email protected]>
rapidsai · Dec 10, 2024 · 83d120d · 83d120d
1 parent c8cc143
commit 83d120d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/source/conf.py b/source/conf.py
@@ -32,6 +32,8 @@
         "rapids_notebooks_container": f"nvcr.io/nvidia/rapidsai/notebooks:{stable_version}-cuda12.5-py3.12",
         "rapids_conda_channels": "-c rapidsai -c conda-forge -c nvidia",
         "rapids_conda_packages": f"rapids={stable_version} python=3.12 cuda-version=12.5",
+        "rapids_pip_index": "https://pypi.nvidia.com",
+        "rapids_pip_version": stable_version,
     },
     "nightly": {
         "rapids_version": f"{nightly_version}-nightly",
@@ -40,6 +42,8 @@
         "rapids_notebooks_container": f"rapidsai/notebooks:{nightly_version + 'a'}-cuda12.5-py3.12",
         "rapids_conda_channels": "-c rapidsai-nightly -c conda-forge -c nvidia",
         "rapids_conda_packages": f"rapids={nightly_version} python=3.12 cuda-version=12.5",
+        "rapids_pip_index": "https://pypi.anaconda.org/rapidsai-wheels-nightly/simple",
+        "rapids_pip_version": f"{nightly_version}.*,>=0.0.0a0",
     },
 }
 rapids_version = (

diff --git a/source/platforms/databricks.md b/source/platforms/databricks.md
@@ -27,11 +27,10 @@ Navigate to the top-left **Workspace** tab and click on your **Home** directory
 set -e
 
 # Install RAPIDS libraries
-pip install --extra-index-url=https://pypi.nvidia.com \
-    "cudf-cu11" \
-    "cuml-cu11" \
-    "dask-cudf-cu11" \
-    "dask-cuda=={{rapids_version}}"
+pip install \
+    --extra-index-url={{rapids_pip_index}} \
+    "cudf-cu11>={{rapids_pip_version}}" "cuml-cu11>={{rapids_pip_version}}" \
+    "dask-cuda>={{rapids_pip_version}}"
 
 ```
 
@@ -51,7 +50,7 @@ The "GPU accelerated" nodes should now be available in the **Node type** dropdow
 
 ![Screenshot of selecting a g4dn.xlarge node type](../images/databricks-choose-gpu-node.png)
 
-Then expand the **Advanced Options** section, open the **Init Scripts** tab and add the file path to the init-script in your Workspace directory starting with `/Users/<user-name>/<script-name>.sh`.
+Then expand the **Advanced Options** section, open the **Init Scripts** tab and enter the file path to the init-script in your Workspace directory starting with `/Users/<user-name>/<script-name>.sh` and click **"Add"**.
 
 ![Screenshot of init script path](../images/databricks-dask-init-script.png)