Install torch models in Dockerfile

This commit is contained in:
Marty Oehme 2023-08-20 13:28:31 +02:00
parent 79a788692f
commit 6e72776fbf
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A

View file

@ -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}