| FROM nvidia/cuda:12.3.2-devel-ubuntu22.04 AS base | |
| # Set environment variables | |
| ENV DEBIAN_FRONTEND=noninteractive | |
| WORKDIR /app | |
| # Install dependencies | |
| RUN apt-get update && apt-get install -y \ | |
| git \ | |
| curl \ | |
| wget \ | |
| python3 \ | |
| python3-pip \ | |
| && rm -rf /var/lib/apt/lists/* | |
| COPY . /app/ | |
| RUN if ! update-alternatives --query python > /dev/null 2>&1; then \ | |
| update-alternatives --install /usr/bin/python python /usr/bin/python3 1; \ | |
| fi | |
| RUN pip install -r /app/requirements.txt | |
| USER root | |
| RUN chmod +x /app/start.sh | |
| EXPOSE 7860 | |
| CMD ["python", "/app/setup.py"] | |