Allow TensorFlow conda packages with CUDA 11.8. (#457)

rapidsai · Nov 27, 2023 · e43537b · e43537b
1 parent 88190d3
commit e43537b
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/_includes/selector.html b/_includes/selector.html
@@ -617,15 +617,14 @@
                     } else {
                         var pkgs_html = this.rapids_meta_pkgs.map(pkg => "<code>" + pkg + "</code>").join(", ");
                     }
-                    notes = [...notes, "The CUDA " + this.active_cuda_ver +
-                        " <code>Standard</code> selection contains the following packages:<br>" + pkgs_html];
+                    notes = [...notes, "The <code>Standard</code> selection contains the following packages:<br>" + pkgs_html];
                 }
 
                 if (this.active_additional_packages.length) {
                     notes = [...notes, "Third-party packages are not tested."];
 
                     if (this.active_additional_packages.includes("TensorFlow")) {
-                        notes = [...notes, "TensorFlow requires CUDA 11.2."];
+                        notes = [...notes, "TensorFlow requires CUDA 11."];
                     }
                 }
 
@@ -677,7 +676,7 @@
             },
             disableUnsupportedCuda(cuda_version) {
                 var isDisabled = false;
-                if (this.active_additional_packages.includes("TensorFlow") && cuda_version !== "11.2") isDisabled = true;
+                if (this.active_additional_packages.includes("TensorFlow") && (cuda_version !== "11.2" && cuda_version !== "11.8")) isDisabled = true;
                 return isDisabled;
             },
             disableUnsupportedPython(python_version) {
@@ -788,7 +787,7 @@
                     return;
                 }
                 this.active_additional_packages = [...this.active_additional_packages, package];
-                if (this.active_additional_packages.includes("TensorFlow") && this.active_cuda_ver !== "11.2") this.active_cuda_ver = "11.2";
+                if (this.active_additional_packages.includes("TensorFlow") && (this.active_cuda_ver !== "11.2" && this.active_cuda_ver !== "11.8")) this.active_cuda_ver = "11.8";
             },
             copyToClipboard() {
                 let range = document.createRange();