fix: use python:3.10-slim base, set HOME=/tmp

Dockerfile

@@ -1,30 +1,22 @@
-FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+FROM python:3.10-slim
 
-ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV MPLCONFIGDIR=/tmp/matplotlib
 ENV PIP_CACHE_DIR=/tmp/pip-cache
+ENV HOME=/tmp
 
-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
+RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
 
 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 torch --index-url https://download.pytorch.org/whl/cu121
+
+RUN pip install --no-cache-dir trl peft accelerate bitsandbytes wandb datasets huggingface_hub transformers
 
-RUN pip install --no-cache-dir \
-    trl peft accelerate bitsandbytes wandb datasets \
-    huggingface_hub torch transformers
+RUN pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps
 
-RUN pip install --no-cache-dir \
-    git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
+RUN pip install --no-cache-dir git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
 
 COPY train.py .