Install torch models in Dockerfile
This commit is contained in:
parent
79a788692f
commit
6e72776fbf
1 changed files with 3 additions and 0 deletions
|
@ -24,6 +24,9 @@ ENV \
|
||||||
RUN curl -sSL https://install.python-poetry.org | python3 -
|
RUN curl -sSL https://install.python-poetry.org | python3 -
|
||||||
ENV PATH="${POETRY_HOME}/bin:${PATH}"
|
ENV PATH="${POETRY_HOME}/bin:${PATH}"
|
||||||
|
|
||||||
|
# installing the large models
|
||||||
|
RUN ltt install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1
|
||||||
|
|
||||||
WORKDIR ${APP_PATH}
|
WORKDIR ${APP_PATH}
|
||||||
COPY ./poetry.lock ./pyproject.toml ./
|
COPY ./poetry.lock ./pyproject.toml ./
|
||||||
COPY ./${APP_NAME} ./${APP_NAME}
|
COPY ./${APP_NAME} ./${APP_NAME}
|
||||||
|
|
Loading…
Reference in a new issue