Create Dockerfile

Dockerfile

@@ -0,0 +1,47 @@
+# Use the NVIDIA CUDA runtime as a base image
+FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
+
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    cmake \
+    build-essential \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    ffmpeg \
+    python3.8 \
+    python3-pip \
+    python3.8-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set Python 3.8 as the default python and pip versions
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1
+RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
+
+# Create a working directory
+WORKDIR /app
+
+# Clone the repository (adjust the URL if needed)
+RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
+
+# Install Python dependencies from requirements.txt
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
+
+# Install basicsr (assuming setup.py is in basicsr directory)
+RUN python basicsr/setup.py develop
+
+# Install additional Python packages
+RUN pip install dlib ffmpeg-python
+
+# Initialize and update Git submodules (if not handled by the clone command)
+RUN git submodule init && git submodule update
+
+# Expose any required ports (if your app uses any)
+# EXPOSE 8000
+
+# Command to run your application
+CMD ["python", "app.py"]