diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index 1df552d..7f038e7 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -1,7 +1,8 @@
-FROM python:3.10
+FROM python:3.12
 
 ARG FACEFUSION_VERSION=3.0.1
 ENV GRADIO_SERVER_NAME=0.0.0.0
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
 
 WORKDIR /facefusion
 
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index fdba166..70a667c 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,12 +1,13 @@
-FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
+FROM nvidia/cuda:12.6.2-cudnn-runtime-ubuntu24.04
 
 ARG FACEFUSION_VERSION=3.0.1
 ENV GRADIO_SERVER_NAME=0.0.0.0
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
 
 WORKDIR /facefusion
 
 RUN apt-get update
-RUN apt-get install python3.10 -y
+RUN apt-get install python3.12 -y
 RUN apt-get install python-is-python3 -y
 RUN apt-get install pip -y
 RUN apt-get install git -y
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 8e1fb22..08b163c 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,12 +1,13 @@
-FROM rocm/dev-ubuntu-22.04:6.2
+FROM rocm/dev-ubuntu-24.04:6.2
 
 ARG FACEFUSION_VERSION=3.0.1
 ENV GRADIO_SERVER_NAME=0.0.0.0
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
 
 WORKDIR /facefusion
 
 RUN apt-get update
-RUN apt-get install python3.10 -y
+RUN apt-get install python3.12 -y
 RUN apt-get install python-is-python3 -y
 RUN apt-get install pip -y
 RUN apt-get install git -y
diff --git a/Dockerfile.tensorrt b/Dockerfile.tensorrt
index a41ee8b..8402c6e 100644
--- a/Dockerfile.tensorrt
+++ b/Dockerfile.tensorrt
@@ -1,12 +1,14 @@
-FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
+FROM nvidia/cuda:12.6.2-cudnn-runtime-ubuntu24.04
 
 ARG FACEFUSION_VERSION=3.0.1
 ENV GRADIO_SERVER_NAME=0.0.0.0
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.12/dist-packages/tensorrt_libs
 
 WORKDIR /facefusion
 
 RUN apt-get update
-RUN apt-get install python3.10 -y
+RUN apt-get install python3.12 -y
 RUN apt-get install python-is-python3 -y
 RUN apt-get install pip -y
 RUN apt-get install git -y
@@ -16,4 +18,3 @@ RUN pip install tensorrt==10.5.0 --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
-
diff --git a/docker-compose.cuda.yml b/docker-compose.cuda.yml
index e5d0cec..d5f8db3 100644
--- a/docker-compose.cuda.yml
+++ b/docker-compose.cuda.yml
@@ -15,4 +15,5 @@ services:
     reservations:
      devices:
      - driver: nvidia
+       count: all
        capabilities: [ gpu, video ]
diff --git a/docker-compose.rocm.yml b/docker-compose.rocm.yml
index efcad47..d9e1b7e 100644
--- a/docker-compose.rocm.yml
+++ b/docker-compose.rocm.yml
@@ -15,4 +15,5 @@ services:
     reservations:
      devices:
      - driver: rocm
+       count: all
        capabilities: [ gpu, video ]
diff --git a/docker-compose.tensorrt.yml b/docker-compose.tensorrt.yml
index a270c7e..22826cc 100644
--- a/docker-compose.tensorrt.yml
+++ b/docker-compose.tensorrt.yml
@@ -1,5 +1,5 @@
 services:
- facefusion-cuda:
+ facefusion-tensorrt:
   build:
    context: .
    dockerfile: Dockerfile.tensorrt
@@ -15,4 +15,5 @@ services:
     reservations:
      devices:
      - driver: nvidia
+       count: all
        capabilities: [ gpu, video ]