fix: back to python:3.10-slim for GPU, fix deceit_env path

Dockerfile

@@ -1,36 +1,34 @@
-FROM nvidia/cuda:12.6.0-cudnn-runtime-ubuntu22.04
+FROM python:3.10-slim
 
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV HOME=/tmp
-ENV DEBIAN_FRONTEND=noninteractive
-ENV PYTHONPATH=/app:/usr/local/lib/python3.10/site-packages
+ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_cache
+ENV PYTHONPATH=/usr/local/lib/python3.10/site-packages
 
-RUN apt-get update && apt-get install -y \
-    python3.10 python3-pip git build-essential \
-    && ln -s /usr/bin/python3.10 /usr/bin/python \
-    && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu124
+RUN pip install --no-cache-dir numpy==1.26.4
+
+RUN pip install --no-cache-dir torch==2.1.2 --index-url https://download.pytorch.org/whl/cu118
 
 RUN pip install --no-cache-dir \
-    transformers accelerate peft trl \
-    bitsandbytes wandb datasets \
-    huggingface_hub matplotlib Pillow
+    transformers==4.36.2 \
+    accelerate==0.25.0 \
+    peft==0.7.1 \
+    trl==0.7.11 \
+    bitsandbytes==0.41.3 \
+    wandb datasets huggingface_hub matplotlib Pillow
 
 RUN pip install --no-cache-dir \
     git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
 
-RUN pip show deceit-env || echo "Package not found"
-RUN python -c "import deceit_env; print(deceit_env.__file__)" || echo "Import failed"
-
-RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/ && \
-    mkdir -p /app/deceit_env/data/
+RUN python -c "import deceit_env; import os; print(os.path.dirname(deceit_env.__file__))"
 
+RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/
 COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
-COPY data/ /app/deceit_env/data/
 COPY data/ /app/data/
 COPY train.py .
 COPY evaluate.py .