From 0e5dfaae37f9d3044011fd3c89dee6c5644dc319 Mon Sep 17 00:00:00 2001
From: Bengt Giger <bgiger@ethz.ch>
Date: Fri, 24 Jan 2025 10:51:22 +0100
Subject: [PATCH] Pytorch in conda

---
 Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 908e094..3cb8793 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 \
-- 
GitLab