diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 09e61c8..722da9d 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -1,6 +1,6 @@ FROM python:3.10 -ARG FACEFUSION_VERSION=2.2.1 +ARG FACEFUSION_VERSION=2.3.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion diff --git a/Dockerfile.cuda b/Dockerfile.cuda index cd4d522..76cb9b4 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -1,6 +1,6 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 -ARG FACEFUSION_VERSION=2.2.1 +ARG FACEFUSION_VERSION=2.3.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion @@ -14,6 +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 --skip-venv +RUN python install.py --torch cuda-11.8 --onnxruntime cuda-11.8 --skip-venv RUN cd /usr/local/lib/python3.10/dist-packages/torch/lib && ln -s libnvrtc-672ee683.so.11.2 libnvrtc.so diff --git a/Dockerfile.rocm b/Dockerfile.rocm index f9faefe..608c026 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,6 +1,6 @@ -FROM rocm/dev-ubuntu-22.04:5.7.1 +FROM rocm/dev-ubuntu-22.04:5.6 -ARG FACEFUSION_VERSION=2.2.1 +ARG FACEFUSION_VERSION=2.3.0 ENV GRADIO_SERVER_NAME=0.0.0.0 WORKDIR /facefusion @@ -14,4 +14,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 rocm --onnxruntime rocm --skip-venv +RUN python install.py --torch rocm-5.6 --onnxruntime rocm-5.6 --skip-venv