Update Dockerfile to install build-essential and update PyTorch version

This commit is contained in:
Quentin Fuxa
2025-08-02 13:08:43 +02:00
parent 8e056cbdf2
commit e4140cd299

View File

@@ -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 . .