diff --git a/backend/dependencies/Dockerfile b/backend/dependencies/Dockerfile
index 8e8723161409d01bb836c1146c756b00f3717b68..9ae337f96a660ecea5d0d52abac0755c8f25472f 100644
--- a/backend/dependencies/Dockerfile
+++ b/backend/dependencies/Dockerfile
@@ -1,6 +1,6 @@
 FROM ubuntu:20.04
 # system packages installation
-
+ARG TARGETPLATFORM
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && \
@@ -49,8 +49,10 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-RUN pip3 install torch torchvision pyvips==2.1.15 cmake==3.21.2
+RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then pip3 install torch torchvision; 
+RUN if [ "$TARGETPLATFORM" = "linux/arm64" ]; then pip3 install torch torchvision -f https://torch.kmtea.eu/whl/stable.html;
 
+RUN pip3 install pyvips==2.1.15 cmake==3.21.2
 
 #Build and install libraw
 WORKDIR /tmp/builds