CalvinBackup/facefusion-docker
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-docker.git synced 2026-04-23 01:25:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

Bump to 2.1.0

Browse Source
This commit is contained in:
henryruhs
2023-12-20 01:04:12 +01:00
parent f986162492
commit 9e7afffefb
3 changed files with 13 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile.cpu
+2 -2
View File
@@ -1,6 +1,6 @@
FROM python:3.10
ARG FACEFUSION_VERSION=2.0.0
ARG FACEFUSION_VERSION=2.1.0
ENV GRADIO_SERVER_NAME=0.0.0.0
WORKDIR /facefusion
@@ -10,4 +10,4 @@ RUN apt-get install curl -y
RUN apt-get install ffmpeg -y
RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
RUN python install.py --torch cpu --onnxruntime default
RUN python install.py --torch cpu --onnxruntime default --skip-venv
Dockerfile.cuda
+3 -2
View File
@@ -1,6 +1,6 @@
FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
ARG FACEFUSION_VERSION=2.0.0
ARG FACEFUSION_VERSION=2.1.0
ENV GRADIO_SERVER_NAME=0.0.0.0
WORKDIR /facefusion
@@ -14,5 +14,6 @@ RUN apt-get install curl -y
RUN apt-get install ffmpeg -y
RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
RUN python install.py --torch cuda --onnxruntime cuda
RUN python install.py --torch cuda --onnxruntime cuda --skip-venv
RUN cd /usr/local/lib/python3.10/dist-packages/torch/lib && ln -s libnvrtc-672ee683.so.11.2 libnvrtc.so
Dockerfile.rocm
+8 -26
View File
@@ -1,35 +1,17 @@
FROM rocm/dev-ubuntu-22.04:5.4.2-complete
FROM rocm/dev-ubuntu-22.04:5.7.1
ARG FACEFUSION_VERSION=2.0.0
ARG ONNXRUNTIME_VERSION=1.15.0
ARG ROCM_VERSION=5.4.2
ARG CMAKE_VERSION=3.27.4
ARG FACEFUSION_VERSION=2.1.0
ENV GRADIO_SERVER_NAME=0.0.0.0
WORKDIR /facefusion
RUN apt-get update
RUN apt-get install python3-dev -y
RUN apt-get install python3.10 -y
RUN apt-get install python-is-python3 -y
RUN apt-get install pip -y
RUN apt-get install git -y
RUN apt-get install curl -y
RUN apt-get install ffmpeg -y
RUN pip install numpy
RUN pip install flake8
RUN pip install packaging
RUN pip install wheel
RUN curl -L https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-x86_64.tar.gz | tar xz -C /opt
RUN sudo ln -sf /opt/cmake-${CMAKE_VERSION}-linux-x86_64/bin/cmake /usr/bin/cmake
RUN sudo ln -sf /opt/cmake-${CMAKE_VERSION}-linux-x86_64/bin/ctest /usr/bin/ctest
WORKDIR /onnxruntime
RUN git clone https://github.com/Microsoft/onnxruntime --branch v${ONNXRUNTIME_VERSION} --single-branch .
RUN chmod +x /onnxruntime/tools/ci_build/build.py
RUN /onnxruntime/tools/ci_build/build.py --allow_running_as_root --build_dir build --config Release --build_wheel --update --build --parallel --skip_tests --cmake_extra_defines ONNXRUNTIME_VERSION=${ONNXRUNTIME_VERSION} --use_rocm --rocm_home /opt/rocm
WORKDIR /facefusion
RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
RUN pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/rocm${ROCM_VERSION}
RUN pip uninstall onnxruntime -y
RUN pip install /onnxruntime/build/Release/dist/*.whl
RUN python install.py --torch rocm --onnxruntime rocm --skip-venv