Rathalos commited on
Commit
8bb3561
·
verified ·
1 Parent(s): 690ac72
Files changed (1) hide show
  1. Dockerfile +14 -1
Dockerfile CHANGED
@@ -1,5 +1,7 @@
1
  # Utilizza l'immagine base NVIDIA CUDA con supporto GPU
2
  FROM huggingface/transformers-pytorch-gpu:latest
 
 
3
  # Installa Python, pip e altre dipendenze di sistema
4
  RUN apt-get update && apt-get install -y \
5
  python3 \
@@ -18,7 +20,7 @@ RUN pip install torch transformers datasets
18
  RUN mkdir -p /app/.cache
19
 
20
  # Imposta le variabili d'ambiente per la cache e altre configurazioni
21
- ENV TRANSFORMERS_CACHE=/app/.cache
22
  ENV HUGGINGFACE_TOKEN=hf_1234567890abcdef1234567890abcdef12345678
23
  ENV HUGGINGFACE_USER=Rathalos
24
  ENV HUGGINGFACE_REPO=training_incite
@@ -29,3 +31,14 @@ WORKDIR /app
29
 
30
  # Comando per eseguire lo script di training
31
  CMD ["python", "train_model.py"]
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Utilizza l'immagine base NVIDIA CUDA con supporto GPU
2
  FROM huggingface/transformers-pytorch-gpu:latest
3
+ # Installa Python, pip e altre dipendenze di sistema
4
+
5
  # Installa Python, pip e altre dipendenze di sistema
6
  RUN apt-get update && apt-get install -y \
7
  python3 \
 
20
  RUN mkdir -p /app/.cache
21
 
22
  # Imposta le variabili d'ambiente per la cache e altre configurazioni
23
+ ENV HF_HOME=/app/.cache
24
  ENV HUGGINGFACE_TOKEN=hf_1234567890abcdef1234567890abcdef12345678
25
  ENV HUGGINGFACE_USER=Rathalos
26
  ENV HUGGINGFACE_REPO=training_incite
 
31
 
32
  # Comando per eseguire lo script di training
33
  CMD ["python", "train_model.py"]
34
+ Struttura dei File Locali
35
+ Assicurati che i tuoi file locali siano organizzati come segue:
36
+
37
+ bash
38
+ Copia codice
39
+ /project_directory
40
+ Dockerfile
41
+ requirements.txt
42
+ train_model.py
43
+ /data
44
+ text_file.txt