CalvinBackup/facefusion-docker
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-docker.git synced 2026-08-12 23:10:34 +02:00
Code Issues Packages Projects Releases Wiki Activity

Modernize dockerfiles

Browse Source
This commit is contained in:
henryruhs
2024-11-05 00:21:17 +01:00
parent 6bd85d3cd6
commit 5137e2b1d8
7 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile.cpu
+2 -1
View File
@@ -1,7 +1,8 @@
FROM python:3.10 FROM python:3.12
ARG FACEFUSION_VERSION=3.0.1 ARG FACEFUSION_VERSION=3.0.1
ENV GRADIO_SERVER_NAME=0.0.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0
ENV PIP_BREAK_SYSTEM_PACKAGES=1
WORKDIR /facefusion WORKDIR /facefusion
Dockerfile.cuda
+3 -2
View File
@@ -1,12 +1,13 @@
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 FROM nvidia/cuda:12.6.2-cudnn-runtime-ubuntu24.04
ARG FACEFUSION_VERSION=3.0.1 ARG FACEFUSION_VERSION=3.0.1
ENV GRADIO_SERVER_NAME=0.0.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0
ENV PIP_BREAK_SYSTEM_PACKAGES=1
WORKDIR /facefusion WORKDIR /facefusion
RUN apt-get update RUN apt-get update
RUN apt-get install python3.10 -y RUN apt-get install python3.12 -y
RUN apt-get install python-is-python3 -y RUN apt-get install python-is-python3 -y
RUN apt-get install pip -y RUN apt-get install pip -y
RUN apt-get install git -y RUN apt-get install git -y
Dockerfile.rocm
+3 -2
View File
@@ -1,12 +1,13 @@
FROM rocm/dev-ubuntu-22.04:6.2 FROM rocm/dev-ubuntu-24.04:6.2
ARG FACEFUSION_VERSION=3.0.1 ARG FACEFUSION_VERSION=3.0.1
ENV GRADIO_SERVER_NAME=0.0.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0
ENV PIP_BREAK_SYSTEM_PACKAGES=1
WORKDIR /facefusion WORKDIR /facefusion
RUN apt-get update RUN apt-get update
RUN apt-get install python3.10 -y RUN apt-get install python3.12 -y
RUN apt-get install python-is-python3 -y RUN apt-get install python-is-python3 -y
RUN apt-get install pip -y RUN apt-get install pip -y
RUN apt-get install git -y RUN apt-get install git -y
Dockerfile.tensorrt
+4 -3
View File
@@ -1,12 +1,14 @@
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 FROM nvidia/cuda:12.6.2-cudnn-runtime-ubuntu24.04
ARG FACEFUSION_VERSION=3.0.1 ARG FACEFUSION_VERSION=3.0.1
ENV GRADIO_SERVER_NAME=0.0.0.0 ENV GRADIO_SERVER_NAME=0.0.0.0
ENV PIP_BREAK_SYSTEM_PACKAGES=1
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.12/dist-packages/tensorrt_libs
WORKDIR /facefusion WORKDIR /facefusion
RUN apt-get update RUN apt-get update
RUN apt-get install python3.10 -y RUN apt-get install python3.12 -y
RUN apt-get install python-is-python3 -y RUN apt-get install python-is-python3 -y
RUN apt-get install pip -y RUN apt-get install pip -y
RUN apt-get install git -y RUN apt-get install git -y
@@ -16,4 +18,3 @@ RUN pip install tensorrt==10.5.0 --extra-index-url https://pypi.nvidia.com
RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch . RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
RUN python install.py --onnxruntime cuda --skip-conda RUN python install.py --onnxruntime cuda --skip-conda
docker-compose.cuda.yml
+1
View File
@@ -15,4 +15,5 @@ services:
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
count: all
capabilities: [ gpu, video ] capabilities: [ gpu, video ]
docker-compose.rocm.yml
+1
View File
@@ -15,4 +15,5 @@ services:
reservations: reservations:
devices: devices:
- driver: rocm - driver: rocm
count: all
capabilities: [ gpu, video ] capabilities: [ gpu, video ]
docker-compose.tensorrt.yml
+2 -1
View File
@@ -1,5 +1,5 @@
services: services:
facefusion-cuda: facefusion-tensorrt:
build: build:
context: . context: .
dockerfile: Dockerfile.tensorrt dockerfile: Dockerfile.tensorrt
@@ -15,4 +15,5 @@ services:
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
count: all
capabilities: [ gpu, video ] capabilities: [ gpu, video ]