diff --git a/NEWS.md b/NEWS.md index fa886cae9f..33b53c5de8 100644 --- a/NEWS.md +++ b/NEWS.md @@ -1,5 +1,9 @@ # keras3 (development version) +- On Linux, `use_backend("jax", gpu = TRUE)` no longer overrides + `XLA_PYTHON_CLIENT_MEM_FRACTION`, preserving the JAX default or a + user-provided value. + # keras3 1.5.1 - `use_backend("jax")` on macOS now defaults to `gpu = FALSE`, so `jax-metal` diff --git a/R/install.R b/R/install.R index 7a075376c6..1ec3634cbc 100644 --- a/R/install.R +++ b/R/install.R @@ -245,7 +245,6 @@ use_backend <- function(backend, gpu = NA) { gpu <- has_gpu() if (gpu) { - Sys.setenv("XLA_PYTHON_CLIENT_MEM_FRACTION" = "1.00") py_require(c("jax[cuda12]!=0.6.1")) } else { py_require(c("jax[cpu]"))