diff --git a/Dockerfile b/Dockerfile
index 908e094de7fb7fa44b3097a88ba2d0c915ac4d90..3cb8793fb7417473000b5f646f55205357ed7ae8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -71,8 +71,7 @@ RUN Rscript >/dev/null 2>&1 -e "install.packages(pkgs=c( \
 # Torch: failed to get filesystem from image: mkdir /opt/conda/lib/python3.12/site-packages/torch/include/ATen: file exists
 # https://github.com/GoogleContainerTools/kaniko/issues/3091
 
-RUN PIP_PROXY=http://proxy.ethz.ch:3128 pip3 install --proxy=http://proxy.ethz.ch:3128 --default-timeout=100 torch && \
-  conda config --set default_threads 4 && \
+RUN conda config --set default_threads 4 && \
   mamba install -q -c defaults -c conda-forge -c bioconda -c bokeh \
   arviz \
   autogluon \
@@ -106,6 +105,7 @@ RUN PIP_PROXY=http://proxy.ethz.ch:3128 pip3 install --proxy=http://proxy.ethz.c
   pymrio \
   pyteomics \
   pytesseract \
+  pytorch::pytorch \
   pyspellchecker \
   rdkit \
   root \