You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

28 lines
1.4 KiB

4 years ago
  1. FROM python:3.7
  2. WORKDIR /project
  3. RUN apt-get update && apt-get install -y git libatlas-base-dev gfortran python3-grpcio python3-scipy python3-numpy python3-sklearn python3-llvmlite python3-cffi python3-audioread python3-numba
  4. ENV PYTHONPATH "${PYTHONPATH}":"/usr/lib/python3/dist-packages/"
  5. RUN git clone -q --recursive https://github.com/NVIDIA/tacotron2.git
  6. RUN git clone https://github.com/alokprasad/fastspeech_squeezewave
  7. RUN rm -rf tacotron2/waveglow
  8. RUN mv fastspeech_squeezewave/SqueezeWave tacotron2/waveglow
  9. RUN rm -rf fastspeech_squeezewave
  10. #COPY MLPTTS .
  11. #COPY squeezewave_dict.pt .
  12. COPY synthesize.py .
  13. COPY merged.dict_1.1.txt .
  14. ARG TARGETPLATFORM
  15. ARG BUILDPLATFORM
  16. RUN echo "I am running on $BUILDPLATFORM, building for $TARGETPLATFORM" > /log
  17. COPY arm_wheels/tensorflow-1.14.0-cp37-none-linux_aarch64.whl .
  18. COPY arm_wheels/torch-1.4.0a0+7f73f1d-cp37-cp37m-linux_aarch64.whl .
  19. #RUN if [ "x$TARGETPLATFORM" = "xlinux/arm64" ] ; then pip install tensorflow-1.14.0-cp37-none-linux_aarch64.whl; rm tensorflow-1.14.0-cp37-none-linux_aarch64.whl; else pip install tensorflow==1.14.0; fi
  20. RUN if [ "x$TARGETPLATFORM" = "xlinux/arm64" ] ; then pip install torch-1.4.0a0+7f73f1d-cp37-cp37m-linux_aarch64.whl; rm torch-1.4.0a0+7f73f1d-cp37-cp37m-linux_aarch64.whl; else pip install torch; fi
  21. #RUN pip install tensorflow==1.14.0
  22. #COPY tensorflow-1.14.0-cp37-none-linux_armv7l.whl .
  23. RUN python3 -m pip install --user librosa==0.6.0 unidecode