pip install nvidia-cudnn-cu11 by default

rstudio · Aug 1, 2023 · 56a7dbc · 56a7dbc
1 parent ddd4fea
commit 56a7dbc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/install.R b/R/install.R
@@ -148,6 +148,9 @@ function(method = c("auto", "virtualenv", "conda"),
     as.character(extra_packages)
   ))
 
+  if(is_linux() && version %in% c("default", "release", "2.13"))
+    packages <- c("nvidia-cudnn-cu11==8.6.*")
+
   if (isTRUE(new_env)) {
 
     if (method %in% c("auto", "virtualenv") &&