diff --git a/api/Containerfile.cpu.buster b/api/Containerfile.cpu.buster
index 44c3e6c2..cf5af69a 100644
--- a/api/Containerfile.cpu.buster
+++ b/api/Containerfile.cpu.buster
@@ -14,11 +14,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
 
 RUN pip3 install pip --upgrade
 
-RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu --no-cache-dir --ignore-installed
+COPY requirements/ /onnx-web/api/requirements/
 
-COPY requirements.txt /onnx-web/api/requirements.txt
-
-RUN pip3 install -r requirements.txt --no-cache-dir
+RUN pip3 install -r requirements/base.txt -r requirements/cpu.txt --no-cache-dir
 
 COPY onnx_web/ /onnx-web/api/onnx_web/
 
diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu
index 8c2ec159..9c1e5c22 100644
--- a/api/Containerfile.cuda.ubuntu
+++ b/api/Containerfile.cuda.ubuntu
@@ -16,13 +16,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
 
 RUN pip3 install pip --upgrade
 
-RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117 --no-cache-dir --ignore-installed
+COPY requirements/ /onnx-web/api/requirements/
 
-COPY requirements.txt /onnx-web/api/requirements.txt
-
-RUN pip3 install -r requirements.txt --no-cache-dir
-
-RUN pip3 install onnxruntime-gpu --no-cache-dir
+RUN pip3 install -r requirements/base.txt -r requirements/nvidia.txt --no-cache-dir
 
 COPY onnx_web/ /onnx-web/api/onnx_web/
 
diff --git a/api/Containerfile.directml.buster b/api/Containerfile.directml.buster
index b51610cb..d4183c8e 100644
--- a/api/Containerfile.directml.buster
+++ b/api/Containerfile.directml.buster
@@ -10,15 +10,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
 
 RUN pip3 install pip --upgrade
 
-RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu --no-cache-dir --ignore-installed
+COPY requirements/ /onnx-web/api/requirements/
 
-RUN pip3 install onnxruntime-directml --force-reinstall
-
-RUN pip3 install --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ ort-nightly-directml --force-reinstall
-
-COPY requirements.txt /onnx-web/api/requirements.txt
-
-RUN pip3 install -r requirements.txt --no-cache-dir
+RUN pip3 install -r requirements/base.txt -r requirements/amd-windows-nightly.txt --no-cache-dir
 
 COPY onnx_web/ /onnx-web/api/onnx_web/
 
diff --git a/api/Containerfile.rocm.ubuntu b/api/Containerfile.rocm.ubuntu
index f24f6720..221faa8f 100644
--- a/api/Containerfile.rocm.ubuntu
+++ b/api/Containerfile.rocm.ubuntu
@@ -14,15 +14,9 @@ ENV PATH="/onnx-web/api/onnx_env/bin:$PATH"
 
 RUN pip3 install pip --upgrade
 
-COPY requirements.txt /onnx-web/api/requirements.txt
+COPY requirements/ /onnx-web/api/requirements/
 
-RUN pip3 install -r requirements.txt --no-cache-dir
-
-WORKDIR /ort
-
-RUN wget https://download.onnxruntime.ai/onnxruntime_training-1.14.0.dev20230123001%2Brocm54-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
-
-RUN pip3 install /ort/*.whl --ignore-installed
+RUN pip3 install -r requirements/base.txt -r requirements/amd-linux.txt --no-cache-dir
 
 WORKDIR /onnx-web/api