Skip to content

Start defaulting to CUDA 12 and Python 3.12 in examples #377

Closed
@jameslamb

Description

@jameslamb

Description

RAPIDS supports CUDA 11.x and 12.x, and has for a couple of releases.

Throughout the examples in this repo, most examples default to CUDA 11. For example:

versions = {
"stable": {
"rapids_version": stable_version,
"rapids_container": f"nvcr.io/nvidia/rapidsai/base:{stable_version}-cuda11.8-py3.10",
"rapids_notebooks_container": f"nvcr.io/nvidia/rapidsai/notebooks:{stable_version}-cuda11.8-py3.10",
"rapids_conda_channels": "-c rapidsai -c conda-forge -c nvidia",
"rapids_conda_packages": f"rapids={stable_version} python=3.10 cuda-version=11.8",
},
"nightly": {
"rapids_version": f"{nightly_version}-nightly",
"rapids_container": f"rapidsai/base:{nightly_version + 'a'}-cuda11.8-py3.10",
"rapids_notebooks_container": f"rapidsai/notebooks:{nightly_version + 'a'}-cuda11.8-py3.10",
"rapids_conda_channels": "-c rapidsai-nightly -c conda-forge -c nvidia",
"rapids_conda_packages": f"rapids={nightly_version} python=3.10 cuda-version=11.8",
},
}

We should update the defaults in these docs to encourage the use of CUDA 12.

Benefits of this work

  • nudges RAPIDS users to move to newer CUDA versions, with all the performance, usability, and other improvements that comes with

Acceptance Criteria

  • all docs published from this repo use CUDA 12 in examples recommending container images and packages, except those where this is a compelling and specific reason to stay on CUDA 11

Approach

Start from here:

versions = {
"stable": {
"rapids_version": stable_version,
"rapids_container": f"nvcr.io/nvidia/rapidsai/base:{stable_version}-cuda11.8-py3.10",
"rapids_notebooks_container": f"nvcr.io/nvidia/rapidsai/notebooks:{stable_version}-cuda11.8-py3.10",
"rapids_conda_channels": "-c rapidsai -c conda-forge -c nvidia",
"rapids_conda_packages": f"rapids={stable_version} python=3.10 cuda-version=11.8",
},
"nightly": {
"rapids_version": f"{nightly_version}-nightly",
"rapids_container": f"rapidsai/base:{nightly_version + 'a'}-cuda11.8-py3.10",
"rapids_notebooks_container": f"rapidsai/notebooks:{nightly_version + 'a'}-cuda11.8-py3.10",
"rapids_conda_channels": "-c rapidsai-nightly -c conda-forge -c nvidia",
"rapids_conda_packages": f"rapids={nightly_version} python=3.10 cuda-version=11.8",
},
}

And then search through the code for other CUDA 11 and CUDA 12 references. Replace them and test the new references.

Notes

N/A

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions