try to correct subprocess spawn and cuda 12.5.1 with cudnn 9...

Dockerfile

@@ -0,0 +1,49 @@
+# Use NVIDIA’s CUDA 12.5.1 with cuDNN 9 on Ubuntu 22.04
+FROM nvidia/cuda:12.5.1-cudnn9-runtime-ubuntu22.04
+
+# Set non-interactive mode for apt-get
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install system packages and add deadsnakes PPA for Python 3.12.9
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    software-properties-common \
+    && add-apt-repository ppa:deadsnakes/ppa -y \
+    && apt-get update && apt-get install -y --no-install-recommends \
+       python3.12 \
+       python3.12-venv \
+       python3.12-dev \
+       python3-pip \
+       git \
+       curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create symlinks for python and pip
+RUN ln -s /usr/bin/python3.12 /usr/local/bin/python && \
+    ln -s /usr/bin/pip3 /usr/local/bin/pip
+
+# Upgrade pip to the latest version
+RUN pip install --upgrade pip
+
+# Install PyTorch with CUDA 12.5 support
+RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu125
+
+# Set the working directory
+WORKDIR /app
+
+# Copy requirements.txt first to leverage Docker caching
+COPY requirements.txt .
+
+# Install other Python dependencies
+RUN pip install -r requirements.txt
+
+# Copy the rest of your repository into the container
+COPY . .
+
+# Set environment variable to mitigate CUDA memory fragmentation
+ENV PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
+
+# Expose the default port for Streamlit
+EXPOSE 8501
+
+# Command to run your Streamlit app
+CMD ["streamlit", "run", "app.py", "--server.enableCORS", "false"]

requirements.txt

@@ -10,8 +10,8 @@ Pillow==10.1.0
 safetensors==0.4.0
 tqdm==4.65.0
 transformers==4.34.1
-torch
-torchvision
+# torch
+# torchvision
 lpips
 # peft
 tensorflow==2.18.0

run_morphing.py

@@ -6,6 +6,9 @@ import argparse
 
 from FILM import process_keyframes
 
+import multiprocessing as mp
+mp.set_start_method("spawn", force=True)
+
 def parse_arguments():
     parser = argparse.ArgumentParser(
         description="Orchestrate DiffMorpher || LCM-LoRa || LCM, and FILM for smooth morphing between two images.")