CablePoleSegmentation / Dockerfile
blitzkrieg0000's picture
Update Dockerfile
4556da0 verified
raw
history blame
373 Bytes
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"]