pytorch问题,运行错误

使用云端服务器跑的一个程序,但是有个错误一直没搞懂
File "/root/miniconda3/lib/python3.8/site-packages/torch_geometric/nn/pool/init.py", line 130, in knn_graph
return torch_cluster.knn_graph(x, k, batch, loop, flow, cosine, 
                                                  num_workers)
AttributeError: 'NoneType' object has no attribute 'knn_graph'
版本为PyTorch 1.9.0

Python 3.8(ubuntu18.04)
Cuda 11.1

直接下载的torch_geometric怎么还有错误呢,求指教啊啊

请问题主解决了吗,我也是这个问题!