.

Dockerfile

@@ -34,9 +34,10 @@ RUN python3 -m pip install numba==0.58.1
 RUN python3 -m pip install torch==2.5.1 torchvision torchaudio
 RUN python3 -m pip install torch_geometric
 #RUN python3 -m pip install  torch_scatter torch_sparse
-RUN python3 -m pip install torch-scatter -f https://data.pyg.org/whl/torch-2.5.1.html
-RUN python3 -m pip install torch-sparse -f https://data.pyg.org/whl/torch-2.5.1.html
-RUN python3 -m pip install torch-cluster -f https://data.pyg.org/whl/torch-2.5.1.html
+#RUN python3 -m pip install torch-scatter -f https://data.pyg.org/whl/torch-2.5.1.html
+#RUN python3 -m pip install torch-sparse -f https://data.pyg.org/whl/torch-2.5.1.html
+#RUN python3 -m pip install torch-cluster -f https://data.pyg.org/whl/torch-2.5.1.html
+RUN python3 -m pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.0+cpu.html
 
 
 RUN python3 -m pip install pytorch-lightning yacs torchmetrics