Fix build
This commit is contained in:
parent
aeccba7c7a
commit
322b44d761
@ -228,13 +228,6 @@ if "KERAS_BACKEND" in os.environ:
|
||||
_BACKEND = _backend
|
||||
|
||||
|
||||
if _BACKEND != "tensorflow":
|
||||
# If we are not running on the tensorflow backend, we should stop tensorflow
|
||||
# from using all available GPU memory. See
|
||||
# https://www.tensorflow.org/guide/gpu#limiting_gpu_memory_growth
|
||||
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
|
||||
|
||||
|
||||
@keras_core_export(
|
||||
[
|
||||
"keras_core.config.backend",
|
||||
|
@ -1,15 +0,0 @@
|
||||
namex
|
||||
black>=22
|
||||
flake8
|
||||
isort
|
||||
pytest
|
||||
pandas
|
||||
absl-py
|
||||
requests
|
||||
h5py
|
||||
protobuf
|
||||
google
|
||||
tensorboard-plugin-profile
|
||||
rich
|
||||
build
|
||||
dm-tree
|
@ -1,18 +0,0 @@
|
||||
# Tensorflow.
|
||||
# Cuda via pip is only on nightly right now.
|
||||
# We will pin a known working version to avoid breakages (nightly breaks often).
|
||||
tf-nightly[and-cuda]==2.14.0.dev20230712
|
||||
|
||||
# Torch.
|
||||
# Pin the version used in colab currently (works with tf cuda version).
|
||||
--extra-index-url https://download.pytorch.org/whl/cu118
|
||||
torch==2.0.1+cu118
|
||||
torchvision==0.15.2+cu118
|
||||
|
||||
# Jax.
|
||||
# Pin the version used in colab currently (works with tf cuda version).
|
||||
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
||||
jax[cuda11_pip]==0.4.10
|
||||
|
||||
# Common deps.
|
||||
-r requirements-common.txt
|
@ -1,13 +1,21 @@
|
||||
# Tensorflow.
|
||||
tensorflow
|
||||
|
||||
# Torch.
|
||||
# TODO: Use Torch CPU, remove after resolving Cuda version differences with TF
|
||||
# TODO: Use Torch CPU
|
||||
# Remove after resolving Cuda version differences with TF
|
||||
torch>=2.0.1+cpu
|
||||
torchvision>=0.15.1
|
||||
|
||||
# Jax.
|
||||
jax[cpu]
|
||||
|
||||
# Common deps.
|
||||
-r requirements-common.txt
|
||||
namex
|
||||
black>=22
|
||||
flake8
|
||||
isort
|
||||
pytest
|
||||
pandas
|
||||
absl-py
|
||||
requests
|
||||
h5py
|
||||
protobuf
|
||||
google
|
||||
tensorboard-plugin-profile
|
||||
rich
|
||||
build
|
||||
dm-tree
|
||||
|
5
setup.py
5
setup.py
@ -23,7 +23,10 @@ def get_version(rel_path):
|
||||
|
||||
HERE = pathlib.Path(__file__).parent
|
||||
README = (HERE / "README.md").read_text()
|
||||
VERSION = get_version("keras_core/__init__.py")
|
||||
if os.path.exists("keras_core/version.py"):
|
||||
VERSION = get_version("keras_core/version.py")
|
||||
else:
|
||||
VERSION = get_version("keras_core/__init__.py")
|
||||
|
||||
setup(
|
||||
name="keras-core",
|
||||
|
Loading…
Reference in New Issue
Block a user