install "tensorflow-metal" on arm macs

rstudio · Aug 16, 2023 · 7dbd9c8 · 7dbd9c8
1 parent 69bbb99
commit 7dbd9c8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/install.R b/R/install.R
@@ -199,6 +199,9 @@ function(method = c("auto", "virtualenv", "conda"),
       python_version <- min(available$version[, 1:2])
   }
 
+  if(is_mac_arm64())
+    packages <- c(packages, "tensorflow-metal")
+
   reticulate::py_install(
     packages       = packages,
     envname        = envname,