From f4e0aef7517e0e8a24ff67faf0b59bdf15167487 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 29 Apr 2025 17:36:37 +0200 Subject: [PATCH] Update images --- Dockerfile.rocm | 2 +- Dockerfile.tensorrt | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 87b297f..c4f2eaa 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,4 +1,4 @@ -FROM rocm/dev-ubuntu-24.04:6.2 +FROM rocm/dev-ubuntu-24.04:6.4 ARG FACEFUSION_VERSION=3.2.0 ENV GRADIO_SERVER_NAME=0.0.0.0 diff --git a/Dockerfile.tensorrt b/Dockerfile.tensorrt index 569c589..13c79bb 100644 --- a/Dockerfile.tensorrt +++ b/Dockerfile.tensorrt @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.8.0-cudnn-runtime-ubuntu24.04 +FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04 ARG FACEFUSION_VERSION=3.2.0 ENV GRADIO_SERVER_NAME=0.0.0.0 @@ -14,7 +14,7 @@ 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 tensorrt==10.8.0.43 --extra-index-url https://pypi.nvidia.com +RUN pip install tensorrt==10.9.0.34 --extra-index-url https://pypi.nvidia.com RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch . RUN python install.py --onnxruntime cuda --skip-conda