From acec5f91902337b2efc142457f8c5c72902c5eee Mon Sep 17 00:00:00 2001 From: Jan Kieseler Date: Tue, 26 Sep 2023 11:53:53 +0200 Subject: [PATCH] fix for torch --- Dockerfile-cpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile-cpu b/Dockerfile-cpu index f584a5d..b337586 100644 --- a/Dockerfile-cpu +++ b/Dockerfile-cpu @@ -24,7 +24,7 @@ RUN apt-get install -y dpkg-dev cmake g++ gcc binutils libx11-dev libxpm-dev lib RUN python3 -m pip install --upgrade pip RUN python3 -m pip install pandas numpy matplotlib MarkupSafe wandb uproot setuptools -RUN python3 -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 +RUN python3 -m pip install --no-cache-dir torch torchvision torchaudio # # # # NO GEANT