Skip to content

Commit

Permalink
simplify docker filename
Browse files Browse the repository at this point in the history
  • Loading branch information
thammegowda committed Mar 16, 2022
1 parent 11c53e3 commit 149415f
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ ENV PATH="/home/rtguser/.local/bin:/usr/local/cuda/bin:${PATH}"
#COPY --chown=rtguser:rtguser . /home/rtguser/rtg/
# && cd /home/rtguser/rtg && pip install --editable . \

RUN pip install --user torch==1.10 flask==1.1.2 uwsgi rtg==0.7 \
RUN pip install --user torch==1.10.2 flask==2.0.3 uwsgi rtg==0.7 \
&& pip cache purge

CMD bash
17 changes: 17 additions & 0 deletions dockers/rtg-model-0.7-600toEng-v2.0.dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# What is this: USC ISI Coral team's MT pipeline
# Authors:
# - Thamme Gowda
# Created : March 15, 2022

FROM tgowda/rtg:0.7-ub20.04-py3.9_tr1.10_cu11.4
# Download pretrained model from google drive

RUN cd /home/rtguser/ && \
curl -O http://rtg.isi.edu/many-eng/models/many-eng-v2.0-rtg600eng1024d_ful74k.tgz && \
tar xvf many-eng-v2.0-rtg600eng1024d_ful74k.tgz --one-top-level=many-eng-v2.0-600toeng --strip-components 1 && \
rm *.tgz

#CMD rtg-serve /home/rtguser/
#CMD python -m rtg.serve /home/rtguser/many-eng-v2.0-600toeng
CMD uwsgi --http 0.0.0.0:6060 --module rtg.serve.app:app --pyargv "/home/rtguser/many-eng-v2.0-600toeng"

0 comments on commit 149415f

Please sign in to comment.