Skip to content

Source install selector from platform support data #802

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
bdice wants to merge 6 commits into
base: main
Choose a base branch
from
Open

Source install selector from platform support data #802

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
9a02dcc
Add 26.08 nightly platform support
bdice Jun 30, 2026
2090519
Update 26.08 nightly platform support
bdice Jun 30, 2026
08fb8bc
Source install selector from platform support data
bdice Jun 30, 2026
2045a34
Merge branch 'main' into codex/selector-platform-support-data
bdice Jun 30, 2026
171dc40
Automate release data maintenance
bdice Jun 30, 2026
9f669fc
Merge branch 'main' into codex/selector-platform-support-data
jakirkham Jun 30, 2026
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
8 changes: 8 additions & 0 deletions .pre-commit-config.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,14 @@ repos:
pass_filenames: false
additional_dependencies:
- pyyaml
- id: validate-platform-support
name: validate-platform-support
entry: ./ci/validate-platform-support.py
language: python
pass_filenames: false
files: ^(ci/validate-platform-support\.py|_data/platform_support\.yml|_data/releases\.json|_includes/selector\.html)$
additional_dependencies:
- pyyaml
- repo: https://github.com/sirosen/texthooks
rev: 0.7.1
hooks:
Expand Down
169 changes: 62 additions & 107 deletions _includes/selector.html
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -249,23 +249,23 @@
</div>
<div class="options-section" x-show="active_method === 'Conda'">
<div class="option-label">ENV. CUDA</div>
<template x-for="version in getSupportedCondaCudaVersions()">
<template x-for="version in getSupportedCudaVersions()">
<div x-on:click="(e) => condaCUDAClickHandler(e, version)"
x-bind:class="{'active': version === active_conda_cuda_ver, 'disabled': disableUnsupportedCuda(version)}"
class="option" x-text="'CUDA ' + getCondaVersionSupport(version)['label']"></div>
</template>
</div>
<div class="options-section" x-show="active_method === 'pip'">
<div class="option-label">System CUDA</div>
<template x-for="version in getSupportedPipCudaVersions()">
<template x-for="version in getSupportedCudaVersions()">
<div x-on:click="(e) => pipCUDAClickHandler(e, version)"
x-bind:class="{'active': version === active_pip_cuda_ver}" class="option"
x-text="'CUDA ' + version"></div>
</template>
</div>
<div class="options-section" x-show="active_method === 'Docker'">
<div class="option-label">Image CUDA</div>
<template x-for="version in getSupportedDockerCudaVersions()">
<template x-for="version in getSupportedCudaVersions()">
<div x-on:click="(e) => dockerCUDAClickHandler(e, version)"
x-bind:class="{'active': version === active_docker_cuda_ver, 'disabled': disableUnsupportedDockerCuda(version)}"
class="option" x-text="'CUDA ' + version"></div>
Expand Down Expand Up @@ -370,26 +370,23 @@
document.addEventListener('alpine:init', () => {
Alpine.data('rapids_selector', () => ({
// default values
active_python_ver: "3.14",
active_conda_cuda_ver: "13",
active_pip_cuda_ver: "13",
active_docker_cuda_ver: "13",
active_python_ver: null,
active_conda_cuda_ver: null,
active_pip_cuda_ver: null,
active_docker_cuda_ver: null,
active_method: "Conda",
active_release: "Stable",
active_img_type: "Base",
active_img_loc: "NGC",
active_packages: ["Standard"],
active_additional_packages: [],

// all possible values
python_vers_stable: ["3.11", "3.12", "3.13", "3.14"],
python_vers_nightly: ["3.11", "3.12", "3.13", "3.14"],
conda_cuda_vers_stable: ["12", "13"],
conda_cuda_vers_nightly: ["12", "13"],
pip_cuda_vers_stable: ["12", "13"],
pip_cuda_vers_nightly: ["12", "13"],
docker_cuda_vers_stable: ["12", "13"],
docker_cuda_vers_nightly: ["12", "13"],
// Release aliases and platform values are rendered from Jekyll data files.
release_versions: {
"Stable": "{{ site.data.releases.stable.version }}",
"Nightly": "{{ site.data.releases.nightly.version }}"
},
platform_support_releases: {{ site.data.platform_support.releases | jsonify }},
methods: ["Conda", "pip", "Docker"],
releases: ["Stable", "Nightly"],
img_loc: ["NGC", "Docker Hub"],
Expand All @@ -400,11 +397,23 @@
additional_packages: ["Graphistry", "JupyterLab", "NetworkX + nx-cugraph", "Plotly Dash", "PyTorch", "TensorFlow", "Xarray-Spatial"],
note_prefix: "<i class='fas fa-info-circle text-blue'></i>",
rapids_meta_pkgs: ["cuDF", "cuML", "cuGraph", "nx-cugraph", "cuxfilter", "cuCIM", "RAFT", "cuVS"],
getStableVersion() {
return "{{ site.data.releases.stable.version }}";
init() {
this.active_python_ver = this.getLatestSupportedVersion(this.getSupportedPythonVersions());
var latest_cuda_version = this.getLatestSupportedVersion(this.getSupportedCudaVersions());
this.active_conda_cuda_ver = latest_cuda_version;
this.active_pip_cuda_ver = latest_cuda_version;
this.active_docker_cuda_ver = latest_cuda_version;
},
getReleaseSupport(release = this.active_release) {
var version = this.release_versions[release];
var support = this.platform_support_releases.find(item => item.version === version);
if (!support) {
throw new Error(`No platform support data found for ${release} release ${version}.`);
}
return support;
},
getNightlyVersion() {
return "{{ site.data.releases.nightly.version }}";
getReleaseVersion(release = this.active_release) {
return this.getReleaseSupport(release).version;
},
getMethodHTML(method) {
var icon_class = "box-open fas";
Expand All @@ -420,8 +429,8 @@
return pkg;
},
getReleaseText(release) {
var version = this.getStableVersion();
if (release.includes("Nightly")) version = this.getNightlyVersion() + "a";
var version = this.getReleaseVersion(release);
if (release === "Nightly") version += "a";
return release + " " + "(" + version + ")";
},
highlightCmd(str) {
Expand All @@ -442,35 +451,28 @@
return pkg_components[0] + channel_pkg_separator + this.highlightPkgOrImg(pkg_components[1]);
},
getCondaVersionSupport(version) {
var cuda_version_info = {
"Stable": {
"12": ["12.2", "12.9"],
"13": ["13.0", "13.2"]
},
"Nightly": {
"12": ["12.2", "12.9"],
"13": ["13.0", "13.3"]
}
};
var bounds = cuda_version_info[this.active_release][version];
// if there is only one bound, we pin the version exactly
if (bounds.length === 1) {
var cuda_support = this.getReleaseSupport().cuda.find(
item => item.major.toString() === version
);
if (!cuda_support) {
throw new Error(`No CUDA ${version} support data found for ${this.active_release}.`);
}
var lower_bound = cuda_support.toolkit_min;
var upper_bound = cuda_support.toolkit_max;
if (lower_bound === upper_bound) {
return {
"label": bounds[0],
"pinning": "=" + bounds[0]
"label": lower_bound,
"pinning": "=" + lower_bound
};
}
// if there are two bounds, we pin the version between the two bounds, inclusive
var lower_bound = bounds[0];
var upper_bound = bounds[1];
return {
"label": lower_bound + " - " + upper_bound,
"pinning": ">=" + lower_bound + ",<=" + upper_bound
};
},
getAdditionalPkgName(pkg) {
// used to pin the version of nx-cugraph for conda install
var rapids_version = this.active_release === "Stable" ? this.getStableVersion() : this.getNightlyVersion();
var rapids_version = this.getReleaseVersion();
var pkg_names = {
"NetworkX + nx-cugraph": `networkx nx-cugraph=${rapids_version}`,
"Plotly Dash": "dash",
Expand All @@ -479,7 +481,7 @@
return (pkg_names[pkg] || pkg).toLowerCase();
},
getCondaCmdHtml() {
var rapids_version = this.active_release === "Stable" ? this.getStableVersion() : this.getNightlyVersion();
var rapids_version = this.getReleaseVersion();
var rapids_channel = this.active_release === "Stable" ? "rapidsai" : "rapidsai-nightly";
var python_version = this.active_python_ver;
var cuda_version_pinning = this.getCondaVersionSupport(this.active_conda_cuda_ver)["pinning"];
Expand Down Expand Up @@ -556,7 +558,7 @@
// This has duplicate code, but makes for easier edits in the future
if (this.active_release === "Stable") {
index_url = `--${this.highlightFlag("extra-index-url")}=https://pypi.nvidia.com`;
var version = this.removeLeadingZeros("{{ site.data.releases.stable.version }}");
var version = this.removeLeadingZeros(this.getReleaseVersion());
cuda_suffix = cuda_suffix + `==${version}.*`;
var libraryToPkg = (pkg) => {
pkg = pkg.toLowerCase();
Expand All @@ -568,7 +570,7 @@
}
else {
index_url = `--${this.highlightFlag("extra-index-url")}=https://pypi.anaconda.org/rapidsai-wheels-nightly/simple`;
var version = this.removeLeadingZeros("{{ site.data.releases.nightly.version }}")
var version = this.removeLeadingZeros(this.getReleaseVersion())
cuda_suffix = cuda_suffix + `>=${version}.0a0,<=${version}`;
var libraryToPkg = (pkg) => {
pkg = pkg.toLowerCase();
Expand Down Expand Up @@ -639,7 +641,7 @@
].filter(Boolean).join("-");

var imgTag = [
(isNightly ? this.getNightlyVersion() + "a" : this.getStableVersion()),
this.getReleaseVersion() + (isNightly ? "a" : ""),
"cuda" + this.active_docker_cuda_ver,
"py" + this.active_python_ver
].join("-");
Expand Down Expand Up @@ -731,28 +733,18 @@
return "Not implemented yet!";
},
getSupportedPythonVersions() {
if (this.active_release === "Stable") {
return this.python_vers_stable;
}
return this.python_vers_nightly;
return this.getReleaseSupport().python;
},
getSupportedDockerCudaVersions() {
if (this.active_release === "Stable") {
return this.docker_cuda_vers_stable;
}
return this.docker_cuda_vers_nightly;
getSupportedCudaVersions() {
return this.getReleaseSupport().cuda.map(item => item.major.toString());
},
getSupportedCondaCudaVersions() {
if (this.active_release === "Stable") {
return this.conda_cuda_vers_stable;
}
return this.conda_cuda_vers_nightly;
getLatestSupportedVersion(versions) {
return versions[versions.length - 1];
},
getSupportedPipCudaVersions() {
if (this.active_release === "Stable") {
return this.pip_cuda_vers_stable;
ensureSupportedVersion(property, supported_versions) {
if (!supported_versions.includes(this[property])) {
this[property] = this.getLatestSupportedVersion(supported_versions);
}
return this.pip_cuda_vers_nightly;
},
disableUnsupportedRelease(release) {
var isDisabled = false;
Expand All @@ -771,7 +763,7 @@
return isDisabled;
},
disableUnsupportedDockerCuda(cuda_version) {
var isDisabled = !this.getSupportedDockerCudaVersions().includes(cuda_version)
var isDisabled = !this.getSupportedCudaVersions().includes(cuda_version)
return isDisabled;
},
disableUnsupportedImgType(type) {
Expand All @@ -797,48 +789,11 @@
releaseClickHandler(e, release) {
if (this.isDisabled(e.target)) return;
this.active_release = release;

/*
It's possible that the active Python version prior to someone changing releases
isn't supported by whatever release was chosen in the selector.

This handles that case, updating to the newest Python version supported by the chosen release.
*/
var supported_python_versions = this.getSupportedPythonVersions();
if (!supported_python_versions.includes(this.active_python_ver)) {
this.active_python_ver = supported_python_versions[supported_python_versions.length - 1];
}

/*
If the selected release doesn't support the selected Docker
CUDA version, update Docker CUDA version to the newest
supported by the selected release.
*/
var supported_docker_cuda_versions = this.getSupportedDockerCudaVersions();
if (!supported_docker_cuda_versions.includes(this.active_docker_cuda_ver)) {
this.active_docker_cuda_ver = supported_docker_cuda_versions[supported_docker_cuda_versions.length - 1];
}

/*
If the selected release doesn't support the selected Conda
CUDA version, update Conda CUDA version to the newest
supported by the selected release.
*/
var supported_conda_cuda_versions = this.getSupportedCondaCudaVersions();
if (!supported_conda_cuda_versions.includes(this.active_conda_cuda_ver)) {
this.active_conda_cuda_ver = supported_conda_cuda_versions[supported_conda_cuda_versions.length - 1];
}

/*
If the selected release doesn't support the selected pip
CUDA version, update pip CUDA version to the newest
supported by the selected release.
*/
var supported_pip_cuda_versions = this.getSupportedPipCudaVersions();
if (!supported_pip_cuda_versions.includes(this.active_pip_cuda_ver)) {
this.active_pip_cuda_ver = supported_pip_cuda_versions[supported_pip_cuda_versions.length - 1];
}

this.ensureSupportedVersion("active_python_ver", this.getSupportedPythonVersions());
var supported_cuda_versions = this.getSupportedCudaVersions();
this.ensureSupportedVersion("active_conda_cuda_ver", supported_cuda_versions);
this.ensureSupportedVersion("active_pip_cuda_ver", supported_cuda_versions);
this.ensureSupportedVersion("active_docker_cuda_ver", supported_cuda_versions);
},
imgTypeClickHandler(e, type) {
if (this.isDisabled(e.target)) return;
Expand Down Expand Up @@ -872,7 +827,7 @@
this.active_packages = ['Standard'];
}
if (method === "pip") {
this.active_pip_cuda_ver = '13';
this.active_pip_cuda_ver = this.getLatestSupportedVersion(this.getSupportedCudaVersions());
}
if (method === "Docker") {
if (this.active_release === 'Nightly') {
Expand Down
Loading
Loading