Actualiser dockerfile.txt

This commit is contained in:
2026-02-23 14:36:45 +00:00
parent ea5fcc89a2
commit 7a2a6faeb4

View File

@ -1,21 +1,23 @@
FROM nvcr.io/nvidia/pytorch:26.01-py3
ENV DEBIAN_FRONTEND=noninteractive
# 1. Système + SSH + Dépendances de rendu (zstd, libgl, libglib)
# 1. Système : Correction des noms de paquets pour Ubuntu Noble
RUN apt-get update && apt-get install -y \
openssh-server ffmpeg libsndfile1 sox git curl zstd \
libgl1-mesa-glx libglib2.0-0 \
libgl1 libglib2.0-0 \
&& apt-get clean && rm -rf /var/lib/apt/lists/*
# Config SSH pour root (nécessaire pour Vast.ai)
# Config SSH pour root
RUN mkdir /var/run/sshd && \
echo 'root:root' | chpasswd && \
sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
# 2. Python & Dépendances
# On installe numpy en premier pour verrouiller la version
RUN pip install --no-cache-dir numpy==1.26.4
RUN pip install --no-cache-dir \
open-webui gradio scipy soundfile julius omegaconf hydra-core \
flashy torchmetrics encodec xformers einops spacy==3.7.6 numpy==1.26.4
flashy torchmetrics encodec xformers einops spacy==3.7.6
RUN pip install --no-cache-dir --no-deps git+https://github.com/facebookresearch/audiocraft.git
@ -29,7 +31,7 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git && \
RUN mkdir -p /root/audiocraft-api
COPY server.py /root/audiocraft-api/server.py
# 5. Script de démarrage avec redirection des logs et sleep infinity
# 5. Script de démarrage
RUN echo '#!/bin/bash\n\
service ssh start\n\
ollama serve > /root/ollama.log 2>&1 & \n\