njlo

mnjpn

Dockerfile

@@ -1,12 +1,17 @@
 # Utilizza l'immagine base ufficiale di Hugging Face per PyTorch con supporto GPU
 FROM huggingface/transformers-pytorch-gpu:latest
 
-# Installa le dipendenze di sistema
+# Aggiorna il sistema e installa le dipendenze di sistema
 RUN apt-get update && apt-get install -y \
+    python3 \
+    python3-pip \
     git \
     wget \
     && rm -rf /var/lib/apt/lists/*
 
+# Imposta un alias per python
+RUN ln -s /usr/bin/python3 /usr/bin/python
+
 # Imposta le variabili d'ambiente per il token e l'utente
 ENV HUGGINGFACE_USER=Rathalos
 ENV HUGGINGFACE_REPO=training_incite
@@ -16,7 +21,7 @@ WORKDIR /app
 
 # Copia il file requirements.txt e installa le dipendenze Python
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip3 install --no-cache-dir -r requirements.txt
 
 # Comando per eseguire lo script di training
 CMD ["python", "train_model.py"]