Update Dockerfile

Dockerfile

@@ -1,23 +1,17 @@
-FROM python:3.11-slim
+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
 
-RUN apt-get update \
- && apt-get install -y --no-install-recommends curl \
- && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y \
+    git \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
 
-RUN useradd -m -u 1000 user
+RUN useradd -m -u 1000 user || true
 ENV USER=user
 ENV HOME=/home/user
 
 WORKDIR /app
-
-COPY pyproject.toml requirements.txt ./
-RUN pip install --no-cache-dir -r requirements.txt
-
 COPY . .
-RUN pip install --no-cache-dir --no-deps .
 
-RUN chmod +x apps/start.sh
-
-EXPOSE 8000
+RUN pip install --no-cache-dir -r requirements.txt
 
-CMD ["./apps/start.sh"]
+CMD ["/bin/bash", "-c", "python -m http.server 7860 & python grpo_train.py"]