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 -
|
||||
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}
|
||||
COPY ./poetry.lock ./pyproject.toml ./
|
||||
COPY ./${APP_NAME} ./${APP_NAME}
|
||||
|
|
Loading…
Reference in a new issue