diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 335f788..34afe1d 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -21,10 +21,14 @@ WORKDIR /app
 # Clone ComfyUI
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
 
+# Create sub dependancy
+RUN grep -vE 'torch|torchsde|torchaudio|torchvision' requirements.txt > req_no_torch.txt
+
+# Install non torch dependant requirements
+RUN pip3 install --no-cache-dir --break-system-packages -r req_no_torch.txt
+
 # Install ComfyUI dependencies
-RUN grep -vE 'torch|torchsde|torchaudio|torchvision' requirements.txt > req_no_torch.txt \
-    && pip3 install --no-deps torchsde \
-    && pip3 install --no-cache-dir -r req_no_torch.txt
+RUN pip3 install --no-deps --break-system-packages torchsde
 
 # Setup Manager
 WORKDIR /app/custom_nodes