Use PyTorch CUDA build. (#523)

rapidsai · Jul 23, 2024 · 7cc3593 · 7cc3593
1 parent 41d220a
commit 7cc3593
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/_includes/selector.html b/_includes/selector.html
@@ -428,7 +428,8 @@
                 var dask_prefix = this.active_release === "Nightly" ? "dask/label/dev::" : "";
                 var pkg_names = {
                     "Plotly Dash": "dash",
-                    "Dask-SQL": dask_prefix + "dask-sql"
+                    "Dask-SQL": dask_prefix + "dask-sql",
+                    "PyTorch": "'pytorch=*=*cuda*'"
                 }
                 return (pkg_names[pkg] || pkg).toLowerCase();
             },