| FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 | |
| WORKDIR /app | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| COPY . . | |
| ENV NVIDIA_VISIBLE_DEVICES=all | |
| CMD ["python", "infer.py"] | |
| FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 | |
| WORKDIR /app | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| COPY . . | |
| ENV NVIDIA_VISIBLE_DEVICES=all | |
| CMD ["python", "infer.py"] | |