Create dockerfile

dockerfile

@@ -0,0 +1,46 @@
+# Use the official CUDA Debian base image from NVIDIA
+FROM nvidia/cuda:11.2.2-devel-ubuntu20.04
+
+# Set environment variables to avoid interactive prompts during package installation
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install Python and system dependencies
+RUN apt-get update && apt-get install -y \
+    python3.10 \
+    python3-pip \
+    build-essential \
+    wget \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create a symlink for python3
+RUN ln -s /usr/bin/python3.10 /usr/bin/python
+
+# Upgrade pip
+RUN python -m pip install --upgrade pip
+
+# Install Python dependencies
+RUN pip install torch==2.0.0+cu118
+    torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 \
+    python-slugify \
+    uuid \
+    peft==0.7.1 \
+    huggingface-hub==0.23.4 \
+    diffusers==0.29.2 \
+    transformers==4.42.3 \
+    accelerate==0.31.0 \
+    safetensors==0.4.3 \
+    prodigyopt==1.0 \
+    hf-transfer==0.1.4 \
+    datasets==2.20.0
+
+# Copy your application code to the container
+COPY . /app
+
+# Set the working directory to /app
+WORKDIR /app
+
+# Specify the command to run your application
+CMD ["python", "app.py"]
+
+# Expose any ports the app will run on
+EXPOSE 7860