torch-geometric version about gaitgraph2

[zhang123-sys]

I use this command python3 gaitgraph_casia_b.py fit --config configs/casia_b.yaml
but it always give me error reply

Traceback (most recent call last):
File "gaitgraph_casia_b.py", line 8, in
from torch_geometric.data import Data
File "/usr/local/lib/python3.8/dist-packages/torch_geometric/init.py", line 4, in
import torch_geometric.data
File "/usr/local/lib/python3.8/dist-packages/torch_geometric/data/init.py", line 1, in
from .data import Data
File "/usr/local/lib/python3.8/dist-packages/torch_geometric/data/data.py", line 9, in
from torch_sparse import SparseTensor
File "/usr/local/lib/python3.8/dist-packages/torch_sparse/init.py", line 19, in
torch.ops.load_library(spec.origin)
File "/usr/local/lib/python3.8/dist-packages/torch/_ops.py", line 104, in load_library
ctypes.CDLL(path)
File "/usr/lib/python3.8/ctypes/init.py", line 373, in init
self._handle = _dlopen(self._name, mode)
OSError: /usr/local/lib/python3.8/dist-packages/torch_sparse/_version_cuda.so: undefined symbol: ZN3c106detail23torchInternalAssertFailEPKcS2_jS2_S2