Experiment using Bazel's pip_install rule to install keras_applications.
PiperOrigin-RevId: 214076591
This commit is contained in:
parent
a2fd40adcc
commit
7229d08f0b
20
WORKSPACE
20
WORKSPACE
@ -9,11 +9,27 @@ http_archive(
|
||||
"https://github.com/bazelbuild/rules_closure/archive/dbb96841cc0a5fb2664c37822803b06dab20c7d1.tar.gz", # 2018-04-13
|
||||
],
|
||||
)
|
||||
|
||||
load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
|
||||
|
||||
closure_repositories()
|
||||
|
||||
http_archive(
|
||||
name = "io_bazel_rules_python",
|
||||
strip_prefix = "rules_python-8b5d0683a7d878b28fffe464779c8a53659fc645",
|
||||
urls = [
|
||||
"https://github.com/bazelbuild/rules_python/archive/8b5d0683a7d878b28fffe464779c8a53659fc645.tar.gz",
|
||||
],
|
||||
)
|
||||
load("@io_bazel_rules_python//python:pip.bzl", "pip_repositories")
|
||||
pip_repositories()
|
||||
|
||||
load("@io_bazel_rules_python//python:pip.bzl", "pip_import")
|
||||
pip_import(
|
||||
name = "pip_deps",
|
||||
requirements = "//tensorflow:requirements.txt",
|
||||
)
|
||||
load("@pip_deps//:requirements.bzl", "pip_install")
|
||||
pip_install()
|
||||
|
||||
# We must check the bazel version before trying to parse any other BUILD
|
||||
# files, in case the parsing of those build files depends on the bazel
|
||||
# version we require here.
|
||||
|
@ -7,6 +7,7 @@ exports_files(["LICENSE"])
|
||||
|
||||
package(default_visibility = ["//visibility:public"])
|
||||
|
||||
load("@pip_deps//:requirements.bzl", "requirement")
|
||||
load("//tensorflow:tensorflow.bzl", "py_test")
|
||||
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
|
||||
|
||||
@ -62,6 +63,7 @@ py_library(
|
||||
":backend",
|
||||
":engine",
|
||||
":layers",
|
||||
requirement("keras_applications"),
|
||||
"//tensorflow/python/saved_model",
|
||||
"//tensorflow/python:training",
|
||||
],
|
||||
|
2
tensorflow/requirements.txt
Normal file
2
tensorflow/requirements.txt
Normal file
@ -0,0 +1,2 @@
|
||||
keras_applications >= 1.0.5
|
||||
keras_preprocessing >= 1.0.3
|
Loading…
Reference in New Issue
Block a user