Create Dockerfile

Dockerfile

@@ -0,0 +1,22 @@
+FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
+
+RUN apt-get update && apt-get install -y git ffmpeg curl
+
+# Install Miniconda
+ENV CONDA_DIR=/opt/conda
+RUN curl -sLo /tmp/miniconda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
+    bash /tmp/miniconda.sh -b -p $CONDA_DIR && rm /tmp/miniconda.sh
+ENV PATH=$CONDA_DIR/bin:$PATH
+
+# Create environment
+RUN conda create -n flovd python=3.10 -y
+SHELL ["conda", "run", "-n", "flovd", "/bin/bash", "-c"]
+
+COPY . /app
+WORKDIR /app
+
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
+
+EXPOSE 7860
+CMD ["conda", "run", "--no-capture-output", "-n", "flovd", "python", "app.py"]