diff --git a/api/Containerfile.alpine b/api/Containerfile.alpine
index 0ec7d42b..ab6f1c9e 100644
--- a/api/Containerfile.alpine
+++ b/api/Containerfile.alpine
@@ -7,6 +7,6 @@ COPY serve.py /onnx-web/serve.py
 
 RUN pip install -r requirements.txt
 
-RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
+# RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
 
 ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]
diff --git a/api/Containerfile.buster b/api/Containerfile.buster
index a606ff5e..6e6ac178 100644
--- a/api/Containerfile.buster
+++ b/api/Containerfile.buster
@@ -7,6 +7,6 @@ COPY serve.py /onnx-web/serve.py
 
 RUN pip install -r requirements.txt
 
-RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
+# RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
 
 ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]