18 lines
409 B
Docker
18 lines
409 B
Docker
|
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
|
||
|
|
|
||
|
|
ENV POETRY_VERSION=1.4.0
|
||
|
|
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
|
||
|
|
|
||
|
|
RUN pip install "poetry==$POETRY_VERSION"
|
||
|
|
|
||
|
|
COPY poetry.lock pyproject.toml /app/
|
||
|
|
RUN poetry export --without-hashes -o requirements.txt
|
||
|
|
|
||
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
||
|
|
|
||
|
|
COPY ./preprocess preprocess
|
||
|
|
COPY ./server server
|