diff --git a/src/dlkcat-gecko/Dockerfile b/src/dlkcat-gecko/Dockerfile
index af0340544..fd240ebb6 100644
--- a/src/dlkcat-gecko/Dockerfile
+++ b/src/dlkcat-gecko/Dockerfile
@@ -1,8 +1,9 @@
 FROM python:3.9-slim
 
 LABEL org.opencontainers.image.source=https://github.com/sysbiochalmers/gecko
-LABEL version="0.1"
+LABEL version="0.1-multiarch"
 LABEL description="Docker image of SysBioChalmers/DKLcat adapted for SysBioChalmers/GECKO version 3"
 
 COPY . .
-RUN pip install --no-cache-dir -r requirements.txt torch@https://download.pytorch.org/whl/cpu/torch-1.9.1%2Bcpu-cp39-cp39-linux_x86_64.whl
\ No newline at end of file
+RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install torch>=1.8.0+cpu --index-url https://download.pytorch.org/whl/cpu
\ No newline at end of file
diff --git a/src/dlkcat-gecko/README.md b/src/dlkcat-gecko/README.md
new file mode 100644
index 000000000..61687fa6a
--- /dev/null
+++ b/src/dlkcat-gecko/README.md
@@ -0,0 +1,5 @@
+Use this command to obtain the multiarch image:
+
+```
+docker buildx build --platform linux/amd64,linux/arm64/v8 --output type=image,name=ghcr.io/sysbiochalmers/dlkcat-gecko,push=true  -t ghcr.io/sysbiochalmers/dlkcat-gecko:0.1-multiarch . 
+```
diff --git a/src/dlkcat-gecko/requirements.txt b/src/dlkcat-gecko/requirements.txt
index 074c9e7ec..612f21743 100644
--- a/src/dlkcat-gecko/requirements.txt
+++ b/src/dlkcat-gecko/requirements.txt
@@ -1,7 +1,6 @@
-torch>=1.8.0
 scikit-learn>=0.23.2
 Biopython==1.78
 rdkit-pypi
 pandas
 SciPy
-NumPy
\ No newline at end of file
+NumPy<2
\ No newline at end of file