Showing
1 changed file
with
23 additions
and
0 deletions
docker/Dockerfile
0 → 100644
| 1 | +FROM nvcr.io/nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04 | ||
| 2 | +LABEL maintainer="nlkey2022@gmail.com" | ||
| 3 | + | ||
| 4 | +RUN DEBIAN_FRONTEND=noninteractive apt-get -qq update \ | ||
| 5 | + && DEBIAN_FRONTEND=noninteractive apt-get -qqy install curl python3-pip git \ | ||
| 6 | + && rm -rf /var/lib/apt/lists/* | ||
| 7 | + | ||
| 8 | +ARG PYTORCH_WHEEL="https://download.pytorch.org/whl/cu101/torch-1.6.0%2Bcu101-cp36-cp36m-linux_x86_64.whl" | ||
| 9 | +ARG ADDED_MODEL="1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4" | ||
| 10 | +ARG DIFF_MODEL="1--gcVVix92_Fp75A-mWH0pJS0ahlni5m" | ||
| 11 | + | ||
| 12 | +RUN git clone https://github.com/graykode/commit-autosuggestions.git /app/commit-autosuggestions \ | ||
| 13 | + && cd /app/commit-autosuggestions && python3 setup.py install | ||
| 14 | + | ||
| 15 | +WORKDIR /app/commit-autosuggestions | ||
| 16 | + | ||
| 17 | +RUN pip3 install ${PYTORCH_WHEEL} gdown | ||
| 18 | +RUN gdown https://drive.google.com/uc?id=${ADDED_MODEL} -O weight/added/ | ||
| 19 | +RUN gdown https://drive.google.com/uc?id=${DIFF_MODEL} -O weight/diff/ | ||
| 20 | + | ||
| 21 | +RUN pip3 install -r requirements.txt | ||
| 22 | + | ||
| 23 | +ENTRYPOINT ["python3", "app.py"] |
-
Please register or login to post a comment