Skip to content

Commit aaa6375

Browse files
authored
Update Dockerfile
1 parent 26e2177 commit aaa6375

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ RUN source ~/.bashrc && \
3434
conda create -n deeptb python=3.9 -c conda-forge -y && \
3535
source activate deeptb && \
3636
pip install torch==2.1.1 && \
37+
pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.1+cpu.html && \
3738
cd ./DeePTB && \
3839
pip install . && \
3940
cd .. && \

0 commit comments

Comments
 (0)