Skip to content

Commit

Permalink
install release tensorflow version by default (not nightly)
Browse files Browse the repository at this point in the history
  • Loading branch information
t-kalinowski committed Jan 4, 2024
1 parent 7558fbd commit c4f7e78
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 11 deletions.
12 changes: 4 additions & 8 deletions R/install.R
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
#' @param envname Name of or path to a Python virtual environment
#' @param extra_packages Additional Python packages to install alongside Keras
#' @param python_version Passed on to `reticulate::virtualenv_starter()`
#' @param backend Which backend(s) to install. Accepted values include `"tensorflow"`, `"jax"` and `"pytorch"`
#' @param backend Which backend(s) to install. Accepted values include `"tensorflow"`, `"jax"` and `"pytorch"`
#' @param gpu whether to install a GPU capable version of the backend.
#' @param restart_session Whether to restart the R session after installing (note this will only occur within RStudio).
#' @param ... reserved for future compatability.
Expand All @@ -14,15 +14,11 @@
#' @export
install_keras <- function(
envname = "r-keras", ...,
extra_packages = c("scipy",
"pandas",
"Pillow",
"pydot",
"ipython",
"tensorflow_datasets"),
extra_packages = c("scipy", "pandas", "Pillow", "pydot", "ipython", "tensorflow_datasets"),
python_version = ">=3.9,<=3.11",
backend = "tensorflow",
# backend = c("tensorflow", "jax"),
backend = "tf-nightly",
# backend = "tf-nightly",
gpu = NA,
restart_session = TRUE) {

Expand Down
4 changes: 2 additions & 2 deletions man/install_keras.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion man/roxygen/meta.R
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@

# this file get's evaled in baseenv()
# Sys.setenv(CUDA_VISIBLE_DEVICES = "")
Sys.setenv(CUDA_VISIBLE_DEVICES = "")

# register fake @tether tag parser for roxygen2
local({
Expand Down

0 comments on commit c4f7e78

Please sign in to comment.