Error for running gun_custom (package missing)  #13

python3 gcn_custom.py --n-hidden=32
Traceback (most recent call last):
File "/home/jinyaox/ge-spmm/pytorch-custom/gcn_custom.py", line 6, in
from torch_geometric.datasets import Planetoid
File "/home/jinyaox/.local/lib/python3.10/site-packages/torch_geometric-1.4.3-py3.10.egg/torch_geometric/init.py", line 2, in
File "/home/jinyaox/.local/lib/python3.10/site-packages/torch_geometric-1.4.3-py3.10.egg/torch_geometric/nn/init.py", line 2, in
File "/home/jinyaox/.local/lib/python3.10/site-packages/torch_geometric-1.4.3-py3.10.egg/torch_geometric/nn/data_parallel.py", line 5, in
File "/home/jinyaox/.local/lib/python3.10/site-packages/torch_geometric-1.4.3-py3.10.egg/torch_geometric/data/init.py", line 5, in
File "/home/jinyaox/.local/lib/python3.10/site-packages/torch_geometric-1.4.3-py3.10.egg/torch_geometric/data/dataloader.py", line 5, in
ImportError: cannot import name 'container_abcs' from 'torch._six' (/home/jinyaox/.local/lib/python3.10/site-packages/torch/_six.py)

I followed all steps