FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime | |
RUN useradd -m -u 1000 user | |
WORKDIR /app | |
COPY --chown=user ./requirements.txt requirements.txt | |
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y | |
RUN pip install --no-cache-dir gradio opencv-python pandas ultralytics onnx onnxruntime | |
COPY --chown=user . /app | |
EXPOSE 1071 | |
CMD ["python", "/app/UI/Main.py"] |