|
358 | 358 | document.addEventListener('alpine:init', () => {
|
359 | 359 | Alpine.data('rapids_selector', () => ({
|
360 | 360 | // 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", |
364 | 364 | active_method: "Conda",
|
365 | 365 | active_release: "Stable",
|
366 | 366 | active_img_type: "Base",
|
|
377 | 377 | img_loc: ["NGC", "Dockerhub"],
|
378 | 378 | img_types: ["Base", "Notebooks"],
|
379 | 379 | 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"], |
382 | 382 | additional_packages: ["Dask-SQL", "JupyterLab", "Plotly Dash", "Graphistry", "TensorFlow", "Xarray-Spatial", "PyTorch"],
|
383 | 383 | note_prefix: "<i class='fas fa-info-circle text-blue'></i>",
|
384 | 384 | rapids_meta_pkgs: ["cuDF", "cuML", "cuGraph", "cuSpatial", "cuProj", "cuxfilter", "cuCIM", "RAFT"],
|
|
465 | 465 | pkgs.push("pylibraft");
|
466 | 466 | pkgs.push("raft-dask");
|
467 | 467 | }
|
| 468 | + if (pkgs.includes("cuVS")) { |
| 469 | + pkgs = pkgs.filter(pkg => pkg !== "cuVS"); |
| 470 | + pkgs.push("cuvs"); |
| 471 | + } |
468 | 472 |
|
469 | 473 |
|
470 | 474 | var pkgs_vers = pkgs.filter(pkg => pkg !== "Choose Specific Packages").map(pkg => this.highlightPkgOrImg(pkg) + "=" + rapids_version + " ").join("").toLowerCase();
|
|
511 | 515 | pkg = pkg.toLowerCase();
|
512 | 516 | if (pkg === "cuspatial/cuproj") return ["cuspatial" + cuda_suffix, "cuproj" + cuda_suffix];
|
513 | 517 | if (pkg === "raft") return ["pylibraft" + cuda_suffix, "raft-dask" + cuda_suffix];
|
| 518 | + if (pkg === "cuvs") return ["cuvs" + cuda_suffix]; |
514 | 519 | return [pkg + cuda_suffix];
|
515 | 520 | }
|
516 | 521 | }
|
|
522 | 527 | pkg = pkg.toLowerCase();
|
523 | 528 | if (pkg === "cuspatial/cuproj") return ["\"cuspatial" + cuda_suffix, "\"cuproj" + cuda_suffix];
|
524 | 529 | if (pkg === "raft") return ["\"pylibraft" + cuda_suffix, "\"raft-dask" + cuda_suffix];
|
| 530 | + if (pkg === "cuvs") return ["\"cuvs" + cuda_suffix, "\"pylibraft" + cuda_suffix]; |
525 | 531 | return ["\"" + pkg + cuda_suffix];
|
526 | 532 | }
|
527 | 533 | }
|
|
629 | 635 | var notes = [];
|
630 | 636 | var install_location_notes = "RAPIDS pip packages are hosted by NVIDIA<br>"
|
631 | 637 | 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>']; |
633 | 639 |
|
634 | 640 | return notes.map(note => this.note_prefix + " " + note);
|
635 | 641 | },
|
|
668 | 674 | var isDisabled = false;
|
669 | 675 | if (this.active_additional_packages.includes("TensorFlow") && (cuda_version !== "12.0")) isDisabled = true;
|
670 | 676 | if (this.active_method === "Docker" && cuda_version < "11.8") isDisabled = true;
|
671 |
| - if (this.active_release != "Nightly" && cuda_version === "12.2") isDisabled = true; |
672 | 677 | return isDisabled;
|
673 | 678 | },
|
674 | 679 | disableUnsupportedPython(python_version) {
|
675 | 680 | var isDisabled = false;
|
676 |
| - if (this.active_release != "Nightly" && python_version === "3.11") isDisabled = true; |
677 | 681 | return isDisabled;
|
678 | 682 | },
|
679 | 683 | disableUnsupportedImgType(type) {
|
|
698 | 702 | },
|
699 | 703 | releaseClickHandler(e, release) {
|
700 | 704 | 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"; |
705 | 705 | this.active_release = release;
|
706 | 706 | },
|
707 | 707 | imgTypeClickHandler(e, type) {
|
|
732 | 732 | this.active_packages = ['Standard'];
|
733 | 733 | }
|
734 | 734 | if (method === "pip") {
|
735 |
| - this.active_pip_cuda_ver = '12.0'; |
| 735 | + this.active_pip_cuda_ver = '12'; |
736 | 736 | }
|
737 | 737 | if (method === "Docker") {
|
738 | 738 | if (this.active_release === 'Nightly') {
|
|
0 commit comments