mirror of
https://github.com/facefusion/facefusion-docker.git
synced 2026-04-27 19:45:57 +02:00
henryruhs
36 lines
1.4 KiB
Docker
36 lines
1.4 KiB
Docker
FROM rocm/dev-ubuntu-22.04:5.4.2-complete
|
|
|
|
ARG FACEFUSION_VERSION=1.3.0
|
|
ARG ONNXRUNTIME_VERSION=1.15.0
|
|
ARG ROCM_VERSION=5.4.2
|
|
ARG CMAKE_VERSION=3.27.4
|
|
ENV GRADIO_SERVER_NAME=0.0.0.0
|
|
|
|
RUN apt-get update
|
|
RUN apt-get install python3-dev -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
|