first commit

Dockerfile

@@ -0,0 +1,39 @@
+# Change CUDA and cuDNN version here
+FROM nvidia/cuda:12.4.1-base-ubuntu22.04
+ARG PYTHON_VERSION=3.11
+
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        software-properties-common \
+        wget \
+    && add-apt-repository ppa:deadsnakes/ppa \
+    && apt-get update && apt-get install -y --no-install-recommends \
+        python$PYTHON_VERSION \
+        python$PYTHON_VERSION-dev \
+        python$PYTHON_VERSION-venv \
+    && wget https://bootstrap.pypa.io/get-pip.py -O get-pip.py \
+    && python$PYTHON_VERSION get-pip.py \
+    && rm get-pip.py \
+    && ln -sf /usr/bin/python$PYTHON_VERSION /usr/bin/python \
+    && ln -sf /usr/local/bin/pip$PYTHON_VERSION /usr/local/bin/pip \
+    && python --version \
+    && pip --version \
+    && apt-get purge -y --auto-remove software-properties-common \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+####### Add your own installation commands here #######
+# RUN pip install some-package
+# RUN wget https://path/to/some/data/or/weights
+ RUN apt-get update &&  apt-get install ffmpeg libsm6 libxext6  -y \
+    && apt install -y build-essential g++
+
+WORKDIR /app
+COPY . /app
+
+# Install litserve and requirements
+RUN pip install --no-cache-dir litserve==0.2.16
+RUN pip install -r requirements-GPU.txt
+RUN pip install opencv-python
+EXPOSE 8000
+CMD ["python", "/app/litserver_main.py"]