From 6f6ffff96a893deb901580ee9495b85908286f9d Mon Sep 17 00:00:00 2001 From: Isabel Haide Date: Tue, 10 Oct 2023 09:45:55 +0200 Subject: [PATCH] add torch_geometric and torch-cluster --- Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile b/Dockerfile index 7171a9f..7d74fe0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -25,6 +25,8 @@ RUN python3 -m pip install --upgrade pip RUN python3 -m pip install pandas numpy matplotlib MarkupSafe wandb uproot setuptools awkward-pandas plotly RUN python3 -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 +RUN python3 -m pip install torch_geometric +RUN python3 -m pip install torch-cluster -f https://data.pyg.org/whl/torch-2.1.0+cu118.html # # # # GEANT RUN apt-get -y install libxmu-dev \