27 lines
672 B
Plaintext
27 lines
672 B
Plaintext
# CUDA 11.2.2 + cuDNN 8 on Ubuntu 20.04
|
|
FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONDONTWRITEBYTECODE=1 \
|
|
PYTHONUNBUFFERED=1 \
|
|
LANG=C.UTF-8
|
|
|
|
# Python 3.10 + pip
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
python3.10 python3-pip build-essential \
|
|
&& ln -sf /usr/bin/python3.10 /usr/bin/python \
|
|
&& ln -sf /usr/bin/pip3 /usr/bin/pip \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Python deps
|
|
RUN pip install --no-cache-dir \
|
|
fastapi \
|
|
uvicorn \
|
|
pillow \
|
|
tensorflow==2.19.0 \
|
|
python-multipart
|
|
|
|
EXPOSE 8000
|
|
|
|
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
|