安装pytorch-geometric

382 阅读1分钟

本文已参与 「新人创作礼」 活动,一起开启掘金创作之路。

安装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-sparsetorch-spline-convtorch-scattertorch-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文件离线安装。 这四个库不建议在线,安装。

到这应该就解决了。