Rathalos commited on
Commit
690ac72
·
verified ·
1 Parent(s): 661eec4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -1,6 +1,5 @@
1
  # Utilizza l'immagine base NVIDIA CUDA con supporto GPU
2
- FROM nvidia/cuda:12.1.0-base
3
-
4
  # Installa Python, pip e altre dipendenze di sistema
5
  RUN apt-get update && apt-get install -y \
6
  python3 \
 
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 \