| FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu22.04 | |
| ENV PATH=${CUDA_HOME}/bin:${PATH} | |
| ENV C_INCLUDE_PATH=${CUDA_HOME}/include:${C_INCLUDE_PATH} | |
| ENV LIBRARY_PATH=${CUDA_HOME}/lib64:${LIBRARY_PATH} | |
| RUN apt-get update && apt-get install -y \ | |
| python3-launchpadlib \ | |
| software-properties-common \ | |
| && apt-get update && apt-get install -y \ | |
| python3-pip python3\ | |
| python3-dev tensorrt \ | |
| && apt-get clean \ | |
| && rm -rf /var/lib/apt/lists/* | |
| RUN useradd -m -u 1000 user | |
| USER user | |
| WORKDIR /app | |
| COPY --chown=user ./requirements.txt requirements.txt | |
| RUN pip install --no-cache-dir --upgrade -r requirements.txt | |
| COPY --chown=user . /app | |
| EXPOSE 7860 | |
| CMD ["python3", "entrypoint.py"] |