Dockerfile.export_weights 372 B

123456789101112131415
  1. FROM nvcr.io/nvidia/pytorch:20.01-py3
  2. # Make sure we have all needed modules
  3. RUN python3 -c "import torch; import onnx; import scipy; import numpy; import librosa"
  4. WORKDIR "/workspace"
  5. ADD ./tacotron2 ./tacotron2
  6. ADD ./waveglow ./waveglow
  7. ADD ./common ./common
  8. ADD ./trtis_cpp/scripts ./trtis_cpp/scripts
  9. WORKDIR "/workspace/trtis_cpp"
  10. ENTRYPOINT ["/bin/bash", "-c"]