Skip to content

Commit

Permalink
update dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
BeautyyuYanli committed Jun 25, 2024
1 parent 4aec1e2 commit f9feef5
Showing 1 changed file with 6 additions and 8 deletions.
14 changes: 6 additions & 8 deletions dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
COPY src/ pdm.lock pyproject.toml README.md /app_build/
RUN apt-get update -qq && apt-get install -qqy --no-install-recommends \
make \
build-essential \
Expand All @@ -7,12 +8,9 @@ RUN apt-get update -qq && apt-get install -qqy --no-install-recommends \
ffmpeg \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app_build
COPY src/ pdm.lock pyproject.toml README.md /app_build/
RUN pip install pdm
RUN pdm use $(which python)
RUN pdm sync
RUN pip install pytorch-lightning==2.0.3 nltk>=3.8.1
RUN python -m nltk.downloader averaged_perceptron_tagger && \
python -m nltk.downloader cmudict
RUN pdm cache clear
RUN pip install pdm pytorch-lightning==2.0.3 nltk>=3.8.1 && \
pdm use $(which python) && \
pdm sync && \
pdm cache clear && pip cache purge
RUN python -m nltk.downloader averaged_perceptron_tagger && python -m nltk.downloader cmudict
WORKDIR /workspace

0 comments on commit f9feef5

Please sign in to comment.