18 lines
438 B
Docker
18 lines
438 B
Docker
|
FROM python:3.9
|
||
|
|
|
||
|
|
WORKDIR /code
|
||
|
|
|
||
|
|
COPY ./requirements.txt /code/requirements.txt
|
||
|
|
|
||
|
|
RUN python --version
|
||
|
|
RUN pip3 --version
|
||
|
|
|
||
|
|
RUN pip3 install torch==2.1.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html && \
|
||
|
|
pip3 install setuptools wheel && \
|
||
|
|
pip3 install --no-cache-dir -r /code/requirements.txt -f https://download.pytorch.org/whl/cu111/torch_stable.html
|
||
|
|
|
||
|
|
COPY . /code
|
||
|
|
|
||
|
|
CMD ["fastapi", "run", "server.py", "--port", "8000"]
|
||
|
|
|