From e43537bf60b3a4164895b3aa53714fcf62bb7fa7 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Mon, 27 Nov 2023 09:01:29 -0800
Subject: [PATCH] Allow TensorFlow conda packages with CUDA 11.8. (#457)

---
 _includes/selector.html | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/_includes/selector.html b/_includes/selector.html
index 418e039b182..98fdd54f73f 100644
--- 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();