| # Copyright 2020 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| FROM gcr.io/iree-oss/android@sha256:6212ef312cb4407c36c0437b22ecf7708bb9beda88e07d1e9c2dc4791aeee998 |
| |
| WORKDIR /install-kws |
| |
| ARG KWS_COMMIT=168f27a070dcd4b0ce39a70f9a702608ff10eb44 |
| ENV PYTHONPATH="/local-python:$PYTHONPATH" |
| |
| RUN wget "https://github.com/google-research/google-research/tarball/${KWS_COMMIT?}" \ |
| -O google-research.tar.gz \ |
| && tar --extract --gzip --file=google-research.tar.gz \ |
| --wildcards */kws_streaming --strip-components=1 \ |
| && mkdir /local-python \ |
| && mv kws_streaming /local-python/kws_streaming \ |
| && rm -rf /install-kws |
| |
| WORKDIR / |
| |
| RUN python3 -m pip install --upgrade \ |
| tensorflow==2.7.1 \ |
| keras==2.7.0 \ |
| # JAX. |
| jax \ |
| jaxlib \ |
| flax \ |
| # KWS Dependency |
| tensorflow-model-optimization==0.5.1.dev0 |
