From aaa637515b1d927d3f8925b07cdd50f2e442c5a4 Mon Sep 17 00:00:00 2001
From: Qiangqiang Gu <98570179+QG-phy@users.noreply.github.com>
Date: Fri, 5 Jul 2024 17:46:54 +0800
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile b/Dockerfile
index 3013b04c..c767b1eb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -34,6 +34,7 @@ RUN source ~/.bashrc &&  \
     conda create -n deeptb python=3.9 -c conda-forge -y  && \
     source activate deeptb && \
     pip install torch==2.1.1 && \
+    pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.1+cpu.html && \
     cd ./DeePTB && \
     pip install . && \
     cd ..  && \