from scratch

Dockerfile

@@ -1,32 +1,17 @@
-# Use a lightweight Python image
-FROM python:3-slim
+# Base image with CUDA 12.8, cuDNN 9, PyTorch 2.7.0
+FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-runtime
 
-WORKDIR /usr/src/app
+# Install system tools (optional, like git, wget)
+RUN apt-get update && apt-get install -y \
+    git \
+    && rm -rf /var/lib/apt/lists/*
 
-# Install git and clean up to keep the image small
-RUN apt-get update && \
-    apt-get install -y git wget && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
+# Set working directory
+WORKDIR /app
 
-# Copy everything and set permissions
-COPY . .
-
-# Install required dependencies
-RUN pip install --upgrade pip
+# Copy and install Python dependencies
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Make sft script executable
-COPY sft.py /usr/local/bin/sft
-RUN chmod +x /usr/local/bin/sft
-
-# Set full write permissions for current directory
-RUN chmod -R 777 .
-
-# Set cache environment variables to use current directory
-ENV XDG_CACHE_HOME=./cache
-
-# Start gradio application
-EXPOSE 7860
-ENV GRADIO_SERVER_NAME="0.0.0.0"
-CMD ["python", "app.py"]
+# Default command: open Python shell
+CMD ["python"]