Error load dataset Planetoid/Cora " Could not run 'pyg::index_sort' with arguments from the 'CPU' backend"

[mrkrl]

Hello!
I have just recently started to investigate using PyTorch geometric for some research and was installing and reading the documentation. Installation seemed to work well, but when I got to the first "basic example"

import torch
from torch import Tensor
from torch_geometric.nn import GCNConv
from torch_geometric.datasets import Planetoid

dataset = Planetoid(root='.', name='Cora')

class GCN(torch.nn.Module):
    def __init__(self, in_channels, hidden_channels, out_channels):
        super().__init__()
        self.conv1 = GCNConv(in_channels, hidden_channels)
        self.conv2 = GCNConv(hidden_channels, out_channels)

    def forward(self, x: Tensor, edge_index: Tensor) -> Tensor:
        # x: Node feature matrix of shape [num_nodes, in_channels]
        # edge_index: Graph connectivity matrix of shape [2, num_edges]
        x = self.conv1(x, edge_index).relu()
        x = self.conv2(x, edge_index)
        return x

model = GCN(dataset.num_features, 16, dataset.num_classes)

I obtain an error, but not where I expected, its at the creation of the dataset dataset = Planetoid(root='.', name='CiteSeer')

My gut is its something to do with my installation but can't quite figure it out. The exact error is

NotImplementedError: Could not run 'pyg::index_sort' with arguments from the 'CPU' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'pyg::index_sort' is only available for these backends: [BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMeta, Tracer, AutocastCPU, AutocastCUDA, FuncTorchBatched, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PythonDispatcher].

Currently installed

Torch Version:  2.1.0.dev20230322+cu118
Torch Geometric Version:  2.2.0
NVIDIA-SMI 530.30.02              Driver Version: 530.30.02    CUDA Version: 12.1 

and the exact error log is below also. Any help or suggestions would be greatly appreciated!

Traceback (most recent call last):
  File "/home/markross/work/PyTorchGeomTesting/mantest.py", line 6, in <module>
    dataset = Planetoid(root='.', name='CiteSeer')
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/datasets/planetoid.py", line 93, in __init__
    super().__init__(root, transform, pre_transform)
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/data/in_memory_dataset.py", line 55, in __init__
    super().__init__(root, transform, pre_transform, pre_filter, log)
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/data/dataset.py", line 94, in __init__
    self._process()
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/data/dataset.py", line 211, in _process
    self.process()
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/datasets/planetoid.py", line 151, in process
    data = read_planetoid_data(self.raw_dir, self.name)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/io/planetoid.py", line 80, in read_planetoid_data
    edge_index = edge_index_from_dict(graph, num_nodes=y.size(0))
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_geometric/io/planetoid.py", line 121, in edge_index_from_dict
    edge_index, _ = coalesce_fn(edge_index, None, num_nodes, num_nodes)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_sparse/coalesce.py", line 22, in coalesce
    storage = SparseStorage(row=index[0], col=index[1], value=value,
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_sparse/storage.py", line 156, in __init__
    _, perm = index_sort(idx[1:], max_value)
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch_sparse/utils.py", line 21, in index_sort
    return pyg_lib.ops.index_sort(inputs, max_value)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/pyg_lib/ops/__init__.py", line 257, in index_sort
    return torch.ops.pyg.index_sort(inputs, max_value)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/markross/.pyenv/versions/pytorchGeo311/lib/python3.11/site-packages/torch/_ops.py", line 642, in __call__
    return self._op(*args, **kwargs or {})
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
NotImplementedError: Could not run 'pyg::index_sort' with arguments from the 'CPU' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'pyg::index_sort' is only available for these backends: [BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMeta, Tracer, AutocastCPU, AutocastCUDA, FuncTorchBatched, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PythonDispatcher].

BackendSelect: fallthrough registered at ../aten/src/ATen/core/BackendSelectFallbackKernel.cpp:3 [backend fallback]
Python: registered at ../aten/src/ATen/core/PythonFallbackKernel.cpp:144 [backend fallback]
FuncTorchDynamicLayerBackMode: registered at ../aten/src/ATen/functorch/DynamicLayer.cpp:491 [backend fallback]
Functionalize: registered at ../aten/src/ATen/FunctionalizeFallbackKernel.cpp:290 [backend fallback]
Named: registered at ../aten/src/ATen/core/NamedRegistrations.cpp:7 [backend fallback]
Conjugate: registered at ../aten/src/ATen/ConjugateFallback.cpp:17 [backend fallback]
Negative: registered at ../aten/src/ATen/native/NegateFallback.cpp:19 [backend fallback]
ZeroTensor: registered at ../aten/src/ATen/ZeroTensorFallback.cpp:86 [backend fallback]
ADInplaceOrView: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:63 [backend fallback]
AutogradOther: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:30 [backend fallback]
AutogradCPU: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:34 [backend fallback]
AutogradCUDA: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:42 [backend fallback]
AutogradXLA: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:46 [backend fallback]
AutogradMPS: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:54 [backend fallback]
AutogradXPU: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:38 [backend fallback]
AutogradHPU: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:67 [backend fallback]
AutogradLazy: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:50 [backend fallback]
AutogradMeta: fallthrough registered at ../aten/src/ATen/core/VariableFallbackKernel.cpp:58 [backend fallback]
Tracer: registered at ../torch/csrc/autograd/TraceTypeManual.cpp:294 [backend fallback]
AutocastCPU: fallthrough registered at ../aten/src/ATen/autocast_mode.cpp:506 [backend fallback]
AutocastCUDA: fallthrough registered at ../aten/src/ATen/autocast_mode.cpp:373 [backend fallback]
FuncTorchBatched: registered at ../aten/src/ATen/functorch/LegacyBatchingRegistrations.cpp:815 [backend fallback]
FuncTorchVmapMode: fallthrough registered at ../aten/src/ATen/functorch/VmapModeRegistrations.cpp:28 [backend fallback]
Batched: registered at ../aten/src/ATen/LegacyBatchingRegistrations.cpp:1073 [backend fallback]
VmapMode: fallthrough registered at ../aten/src/ATen/VmapModeRegistrations.cpp:33 [backend fallback]
FuncTorchGradWrapper: registered at ../aten/src/ATen/functorch/TensorWrapper.cpp:210 [backend fallback]
PythonTLSSnapshot: registered at ../aten/src/ATen/core/PythonFallbackKernel.cpp:152 [backend fallback]
FuncTorchDynamicLayerFrontMode: registered at ../aten/src/ATen/functorch/DynamicLayer.cpp:487 [backend fallback]
PythonDispatcher: registered at ../aten/src/ATen/core/PythonFallbackKernel.cpp:148 [backend fallback]

[mrkrl]

Ok, going to write an answer on my own question. When reading back over I noticed my torch version was a dev version 2.1.0.dev20230322+cu118
reinstalling normally the stable pytorch and not from the nightly seemed to work.

Torch Version:  2.0.0+cu118
Torch Geometric Version:  2.2.0

and for future people finding this on google, this meant

python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
instead of
python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118 )

[Srivathsan-Srinivas]

Hmmm...I just encountered the same error. I have the normal CPU installation of torch. Does anyone has any suggestion?

Traceback (most recent call last):
  File "/Users/Sri/exps/graphs/trial_4/src/main.py", line 277, in <module>
    main(get_args())
  File "/Users/Sri/exps/graphs/trial_4/src/main.py", line 236, in main
    trained_anomaly_detector = run_anomaly_detection(args_dict, config, event_id_map_df, input_file, specific_splitted_data)
                               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/exps/graphs/trial_4/src/main.py", line 147, in run_anomaly_detection
    trained_anomaly_detector = ad_obj.train_anomaly_detector()
                               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/exps/graphs/trial_4/src/anomaly_detection.py", line 12, in train_anomaly_detector
    return self.detector.fit(self.graph_data)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/pygod/detector/base.py", line 435, in fit
    loader = NeighborLoader(data,
             ^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/torch_geometric/loader/neighbor_loader.py", line 229, in __init__
    neighbor_sampler = NeighborSampler(
                       ^^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/torch_geometric/sampler/neighbor_sampler.py", line 85, in __init__
    self.colptr, self.row, self.perm = to_csc(
                                       ^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/torch_geometric/sampler/utils.py", line 73, in to_csc
    row, col, perm = sort_csc(row, col, src_node_time, edge_time)
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/torch_geometric/sampler/utils.py", line 23, in sort_csc
    col, perm = index_sort(col)
                ^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/torch_geometric/utils/_index_sort.py", line 32, in index_sort
    return pyg_lib.ops.index_sort(inputs, max_value=max_value)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/pyg_lib/ops/__init__.py", line 336, in index_sort
    return torch.ops.pyg.index_sort(inputs, max_value)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages/torch/_ops.py", line 1116, in __call__
    return self._op(*args, **(kwargs or {}))
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
NotImplementedError: Could not run 'pyg::index_sort' with arguments from the 'CPU' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'pyg::index_sort' is only available for these backends: [MPS, Meta, BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMeta, Tracer, AutocastCPU, AutocastXPU, AutocastMPS, AutocastCUDA, FuncTorchBatched, BatchedNestedTensor, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PreDispatch, PythonDispatcher].

MPS: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/mps/MPSFallback.mm:75 [backend fallback]
Meta: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/MetaFallbackKernel.cpp:23 [backend fallback]
BackendSelect: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/BackendSelectFallbackKernel.cpp:3 [backend fallback]
Python: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:153 [backend fallback]
FuncTorchDynamicLayerBackMode: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/functorch/DynamicLayer.cpp:497 [backend fallback]
Functionalize: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/FunctionalizeFallbackKernel.cpp:349 [backend fallback]
Named: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/NamedRegistrations.cpp:7 [backend fallback]
Conjugate: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/ConjugateFallback.cpp:17 [backend fallback]
Negative: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/native/NegateFallback.cpp:18 [backend fallback]
ZeroTensor: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/ZeroTensorFallback.cpp:86 [backend fallback]
ADInplaceOrView: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:96 [backend fallback]
AutogradOther: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:63 [backend fallback]
AutogradCPU: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:67 [backend fallback]
AutogradCUDA: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:75 [backend fallback]
AutogradXLA: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:79 [backend fallback]
AutogradMPS: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:87 [backend fallback]
AutogradXPU: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:71 [backend fallback]
AutogradHPU: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:100 [backend fallback]
AutogradLazy: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:83 [backend fallback]
AutogradMeta: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:91 [backend fallback]
Tracer: registered at /Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/autograd/TraceTypeManual.cpp:294 [backend fallback]
AutocastCPU: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/autocast_mode.cpp:321 [backend fallback]
AutocastXPU: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/autocast_mode.cpp:463 [backend fallback]
AutocastMPS: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/autocast_mode.cpp:209 [backend fallback]
AutocastCUDA: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/autocast_mode.cpp:165 [backend fallback]
FuncTorchBatched: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/functorch/LegacyBatchingRegistrations.cpp:731 [backend fallback]
BatchedNestedTensor: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/functorch/LegacyBatchingRegistrations.cpp:758 [backend fallback]
FuncTorchVmapMode: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/functorch/VmapModeRegistrations.cpp:27 [backend fallback]
Batched: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/LegacyBatchingRegistrations.cpp:1075 [backend fallback]
VmapMode: fallthrough registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/VmapModeRegistrations.cpp:33 [backend fallback]
FuncTorchGradWrapper: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/functorch/TensorWrapper.cpp:207 [backend fallback]
PythonTLSSnapshot: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:161 [backend fallback]
FuncTorchDynamicLayerFrontMode: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/functorch/DynamicLayer.cpp:493 [backend fallback]
PreDispatch: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:165 [backend fallback]
PythonDispatcher: registered at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:157 [backend fallback]

make: *** [run_ad] Error 1
(ve_3.11.5) Sri@Harnis-MBP trial_4 % python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
Looking in indexes: https://download.pytorch.org/whl/cpu
Requirement already satisfied: torch in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (2.5.1)
Requirement already satisfied: torchvision in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (0.20.1)
Requirement already satisfied: torchaudio in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (2.5.1)
Requirement already satisfied: filelock in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torch) (3.16.1)
Requirement already satisfied: typing-extensions>=4.8.0 in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torch) (4.12.2)
Requirement already satisfied: networkx in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torch) (3.4.2)
Requirement already satisfied: jinja2 in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torch) (3.1.4)
Requirement already satisfied: fsspec in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torch) (2024.10.0)
Requirement already satisfied: sympy==1.13.1 in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torch) (1.13.1)
Requirement already satisfied: mpmath<1.4,>=1.1.0 in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from sympy==1.13.1->torch) (1.3.0)
Requirement already satisfied: numpy in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torchvision) (2.1.3)
Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from torchvision) (11.0.0)
Requirement already satisfied: MarkupSafe>=2.0 in /Users/Sri/.pyenv/versions/ve_3.11.5/lib/python3.11/site-packages (from jinja2->torch) (3.0.2)
(ve_3.11.5) Sri@Harnis-MBP trial_4 %

[Srivathsan-Srinivas]

I think I resolved the above backend problem by lowering my torch version to 2.4.0. Previously, I was at 2.5.1. python3 -m pip install torch==2.4.0

[dqnk]

I have also had this issue just now, but it seems that some of the libraries' versions weren't synced. Have solved it by adding the --upgrade flag to both pip install torch, pip install torch_geometric and those additional optional dependencies on PyG installation guide. The current latest versions torch==2.5.1 and torch_geometric==2.6.1 seem to be working fine together.