Skip to content

Commit d4a1360

Browse files
jarmak-nvbdice
andauthored
24.04 Selector updates (#506)
Co-authored-by: Bradley Dice <[email protected]>
1 parent b6af951 commit d4a1360

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

_includes/selector.html

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -358,9 +358,9 @@
358358
document.addEventListener('alpine:init', () => {
359359
Alpine.data('rapids_selector', () => ({
360360
// default values
361-
active_python_ver: "3.10",
362-
active_cuda_ver: "12.0",
363-
active_pip_cuda_ver: "12.0",
361+
active_python_ver: "3.11",
362+
active_cuda_ver: "12.2",
363+
active_pip_cuda_ver: "12",
364364
active_method: "Conda",
365365
active_release: "Stable",
366366
active_img_type: "Base",
@@ -377,8 +377,8 @@
377377
img_loc: ["NGC", "Dockerhub"],
378378
img_types: ["Base", "Notebooks"],
379379
packages: ["Standard", "Choose Specific Packages"],
380-
additional_pip_packages: ["cuDF", "dask-cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT"],
381-
additional_rapids_packages: ["cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT"],
380+
additional_pip_packages: ["cuDF", "dask-cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"],
381+
additional_rapids_packages: ["cuDF", "cuML", "cuGraph", "cuSpatial/cuProj", "cuxfilter", "cuCIM", "RAFT", "cuVS"],
382382
additional_packages: ["Dask-SQL", "JupyterLab", "Plotly Dash", "Graphistry", "TensorFlow", "Xarray-Spatial", "PyTorch"],
383383
note_prefix: "<i class='fas fa-info-circle text-blue'></i>",
384384
rapids_meta_pkgs: ["cuDF", "cuML", "cuGraph", "cuSpatial", "cuProj", "cuxfilter", "cuCIM", "RAFT"],
@@ -465,6 +465,10 @@
465465
pkgs.push("pylibraft");
466466
pkgs.push("raft-dask");
467467
}
468+
if (pkgs.includes("cuVS")) {
469+
pkgs = pkgs.filter(pkg => pkg !== "cuVS");
470+
pkgs.push("cuvs");
471+
}
468472

469473

470474
var pkgs_vers = pkgs.filter(pkg => pkg !== "Choose Specific Packages").map(pkg => this.highlightPkgOrImg(pkg) + "=" + rapids_version + " ").join("").toLowerCase();
@@ -511,6 +515,7 @@
511515
pkg = pkg.toLowerCase();
512516
if (pkg === "cuspatial/cuproj") return ["cuspatial" + cuda_suffix, "cuproj" + cuda_suffix];
513517
if (pkg === "raft") return ["pylibraft" + cuda_suffix, "raft-dask" + cuda_suffix];
518+
if (pkg === "cuvs") return ["cuvs" + cuda_suffix];
514519
return [pkg + cuda_suffix];
515520
}
516521
}
@@ -522,6 +527,7 @@
522527
pkg = pkg.toLowerCase();
523528
if (pkg === "cuspatial/cuproj") return ["\"cuspatial" + cuda_suffix, "\"cuproj" + cuda_suffix];
524529
if (pkg === "raft") return ["\"pylibraft" + cuda_suffix, "\"raft-dask" + cuda_suffix];
530+
if (pkg === "cuvs") return ["\"cuvs" + cuda_suffix, "\"pylibraft" + cuda_suffix];
525531
return ["\"" + pkg + cuda_suffix];
526532
}
527533
}
@@ -629,7 +635,7 @@
629635
var notes = [];
630636
var install_location_notes = "RAPIDS pip packages are hosted by NVIDIA<br>"
631637
notes = [...notes, install_location_notes,
632-
'pip installation supports Python <code>3.9</code> and <code>3.10</code>. Nightly builds also support <code>3.11</code>.<br>'];
638+
'pip installation supports Python <code>3.9</code>, <code>3.10</code>, and <code>3.11</code>.<br>'];
633639

634640
return notes.map(note => this.note_prefix + " " + note);
635641
},
@@ -668,12 +674,10 @@
668674
var isDisabled = false;
669675
if (this.active_additional_packages.includes("TensorFlow") && (cuda_version !== "12.0")) isDisabled = true;
670676
if (this.active_method === "Docker" && cuda_version < "11.8") isDisabled = true;
671-
if (this.active_release != "Nightly" && cuda_version === "12.2") isDisabled = true;
672677
return isDisabled;
673678
},
674679
disableUnsupportedPython(python_version) {
675680
var isDisabled = false;
676-
if (this.active_release != "Nightly" && python_version === "3.11") isDisabled = true;
677681
return isDisabled;
678682
},
679683
disableUnsupportedImgType(type) {
@@ -698,10 +702,6 @@
698702
},
699703
releaseClickHandler(e, release) {
700704
if (this.isDisabled(e.target)) return;
701-
// Remove below after we support 12.2 in stable
702-
if (this.active_cuda_ver === "12.2" && release === "Stable") this.active_cuda_ver = "12.0";
703-
// Remove below after we support 3.11 in stable
704-
if (this.active_python_ver === "3.11" && release === "Stable") this.active_python_ver = "3.10";
705705
this.active_release = release;
706706
},
707707
imgTypeClickHandler(e, type) {
@@ -732,7 +732,7 @@
732732
this.active_packages = ['Standard'];
733733
}
734734
if (method === "pip") {
735-
this.active_pip_cuda_ver = '12.0';
735+
this.active_pip_cuda_ver = '12';
736736
}
737737
if (method === "Docker") {
738738
if (this.active_release === 'Nightly') {

0 commit comments

Comments
 (0)