Update Dockerfile

deepmodeling · Jul 5, 2024 · aaa6375 · aaa6375
1 parent 26e2177
commit aaa6375
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 ..  && \