Conditional installation of pytorch

backend/dependencies/Dockerfile

 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