mirror of
https://github.com/facefusion/facefusion-docker.git
synced 2026-04-23 01:25:58 +02:00
henryruhs
34 lines
1.3 KiB
Docker
34 lines
1.3 KiB
Docker
FROM rocm/dev-ubuntu-22.04:5.4.2-complete
|
|
|
|
ARG FACEFUSION_VERSION=1.2.1
|
|
ARG ONNXRUNTIME_VERSION=1.15.0
|
|
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/v3.27.4/cmake-3.27.4-linux-x86_64.tar.gz | tar xz -C /opt
|
|
RUN sudo ln -sf /opt/cmake-3.27.4-linux-x86_64/bin/cmake /usr/bin/cmake
|
|
RUN sudo ln -sf /opt/cmake-3.27.4-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/rocm5.4.2
|
|
RUN pip uninstall onnxruntime -y
|
|
RUN pip install /onnxruntime/build/Release/dist/*.whl
|