Update README

applib-sg-1009 · Jun 9, 2023 · b4c5e41 · b4c5e41
1 parent 61eaf98
commit b4c5e41
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -2,6 +2,10 @@
 
 Keras Core is a new multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
 
+## Local installation
+
+Run `python3 pip_build.py --install` from the root directory.
+
 ## Backwards compatibility
 
 Keras Core is intend to work as a drop-in replacement for `tf.keras` (when using the TensorFlow backend).

diff --git a/pip_build.py b/pip_build.py
@@ -22,6 +22,9 @@
 
 import namex
 
+# Needed because importing torch after TF causes the runtime to crash
+import torch  # noqa: F401
+
 package = "keras_core"
 build_directory = "tmp_build_dir"
 dist_directory = "dist"