fix: install deps in Dockerfile build, not runtime

Dockerfile

@@ -1,8 +1,31 @@
-FROM python:3.10-slim
-RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
-WORKDIR /app
-COPY train.py .
+FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV MPLCONFIGDIR=/tmp/matplotlib
+ENV PIP_CACHE_DIR=/tmp/pip-cache
+
+RUN apt-get update && apt-get install -y \
+    python3.10 python3-pip git build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN ln -s /usr/bin/python3.10 /usr/bin/python
+
+WORKDIR /app
+
+# Install all dependencies during build
+RUN pip install --no-cache-dir \
+    "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" \
+    --no-deps
+
+RUN pip install --no-cache-dir \
+    trl peft accelerate bitsandbytes wandb datasets \
+    huggingface_hub torch transformers
+
+RUN pip install --no-cache-dir \
+    git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
+
+COPY train.py .
+
 CMD ["python", "train.py"]

train.py

@@ -1,8 +1,6 @@
-import os, sys, pathlib, subprocess
+import os, sys, pathlib
 import threading
 
-os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
-os.environ["HF_HOME"] = "/tmp/huggingface"
 os.makedirs("/tmp/matplotlib", exist_ok=True)
 os.makedirs("/tmp/huggingface", exist_ok=True)
 from http.server import HTTPServer, BaseHTTPRequestHandler
@@ -24,14 +22,6 @@ health_thread = threading.Thread(target=start_health_server, daemon=True)
 health_thread.start()
 print("Health server started on port 7860")
 
-# Install deps
-subprocess.run([sys.executable, "-m", "pip", "install", "-q",
-    "unsloth", "unsloth_zoo", "nest-asyncio"], check=True)
-subprocess.run([sys.executable, "-m", "pip", "install", "-q",
-    "trl", "peft", "accelerate", "bitsandbytes", "wandb", "datasets",
-    "git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git"],
-    check=True)
-
 import torch, json, re, threading, pathlib
 from unsloth import FastLanguageModel
 from trl import GRPOConfig, GRPOTrainer