安装pytorch-geometric
pip install pytorch-geometric
pip install pytorch-geometric-temporal
完了之后,运行代码,提示
FileNotFoundError: Could not find module 'C:\Softwares\Anaconda\envs\pytorch_spspmm_cuda.pyd
原因是,安装的torch-sparse
、torch-spline-conv
、torch-scatter
、torch-cluster
版本不对。
查看好自己的cuda版本,建议用这个查:
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
之后访问:https://pytorch-geometric.com/whl/
看仔细自己的torch+cuda
版本,点击跳转,下载对应python版本
的whl文件
,离线
安装。
这四个库不建议在线,安装。
到这应该就解决了。