diff --git a/backend/dependencies/Dockerfile b/backend/dependencies/Dockerfile index 9ae337f96a660ecea5d0d52abac0755c8f25472f..4cc9fa08183df936672630f5023d64c071106899 100644 --- a/backend/dependencies/Dockerfile +++ b/backend/dependencies/Dockerfile @@ -48,9 +48,8 @@ RUN apt-get update && \ unzip && \ apt-get clean && \ rm -rf /var/lib/apt/lists/* - -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 if [ "$TARGETPLATFORM" = "linux/arm64" ]; then pip3 install torch torchvision -f https://torch.kmtea.eu/whl/stable.html; else pip3 install torch torchvision; RUN pip3 install pyvips==2.1.15 cmake==3.21.2