Actualiser dockerfile.txt
This commit is contained in:
@ -1,21 +1,23 @@
|
|||||||
FROM nvcr.io/nvidia/pytorch:26.01-py3
|
FROM nvcr.io/nvidia/pytorch:26.01-py3
|
||||||
ENV DEBIAN_FRONTEND=noninteractive
|
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 \
|
RUN apt-get update && apt-get install -y \
|
||||||
openssh-server ffmpeg libsndfile1 sox git curl zstd \
|
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/*
|
&& 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 && \
|
RUN mkdir /var/run/sshd && \
|
||||||
echo 'root:root' | chpasswd && \
|
echo 'root:root' | chpasswd && \
|
||||||
sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
|
sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
|
||||||
|
|
||||||
# 2. Python & Dépendances
|
# 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 \
|
RUN pip install --no-cache-dir \
|
||||||
open-webui gradio scipy soundfile julius omegaconf hydra-core \
|
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
|
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
|
RUN mkdir -p /root/audiocraft-api
|
||||||
COPY server.py /root/audiocraft-api/server.py
|
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\
|
RUN echo '#!/bin/bash\n\
|
||||||
service ssh start\n\
|
service ssh start\n\
|
||||||
ollama serve > /root/ollama.log 2>&1 & \n\
|
ollama serve > /root/ollama.log 2>&1 & \n\
|
||||||
|
|||||||
Reference in New Issue
Block a user