From e4140cd2991fcbd95ef15ac5a6f9c4a02d115e2d Mon Sep 17 00:00:00 2001
From: Quentin Fuxa <quentin.fuxa@gmail.com>
Date: Sat, 2 Aug 2025 13:08:43 +0200
Subject: [PATCH] Update Dockerfile to install build-essential and update
 PyTorch version

---
 Dockerfile | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 72b9119..5e32d5b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,10 +21,12 @@ RUN apt-get update && \
         python3 \
         python3-pip \
         ffmpeg \
-        git && \
+        git \
+        build-essential \
+        python3-dev && \
     rm -rf /var/lib/apt/lists/*
 
-RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
 
 COPY . .