File size: 570 Bytes
592da45
286f32d
 
 
592da45
 
286f32d
 
 
 
5447961
 
286f32d
d2c7288
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04 as base-img
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y python3 ffmpeg  libxext6  libsm6 libgtk2.0-dev pkg-config libgtk2.0-dev  libavcodec-dev python3-opencv libopencv-dev 
RUN apt-get install -y python3-pip
RUN pip3 install torch torchvision  --index-url https://download.pytorch.org/whl/cu118


FROM base-img as app
WORKDIR /app
COPY requirements.txt requirements.txt
COPY assets assets
COPY yologp yologp
RUN pip3 install -r requirements.txt
ENTRYPOINT [ "python3","yologp/gradio_app.py" ]