fix: nvidia cuda base with python3.10 installed

Dockerfile

@@ -1,15 +1,25 @@
-FROM huggingface/transformers-pytorch-gpu:latest
+FROM nvidia/cuda:12.6.0-cudnn-runtime-ubuntu22.04
 
 ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV HOME=/tmp
-ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_cache
+ENV DEBIAN_FRONTEND=noninteractive
+
+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/*
 
 WORKDIR /app
 
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu124
+
+RUN pip install --no-cache-dir \
+    transformers accelerate peft trl \
+    bitsandbytes wandb datasets \
+    huggingface_hub matplotlib Pillow
+
 RUN pip install --no-cache-dir \
-    peft trl bitsandbytes wandb datasets \
-    huggingface_hub matplotlib Pillow \
     git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
 
 RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/