官网安装地址Installation — pytorch_geometric documentation (pytorch-geometric.readthedocs.io)
里面说PyG is available for Python 3.8 to Python 3.11.一开始我直接用的原有的虚拟环境py3.7,所以可能是因为版本问题后续有一个torch-sparse的包无法import,会出现OSError: libcusparse.so.9.2: cannot open shared object file: No such file or directory此类报错
参考写bug日记2:PYTORCH GEOMETRIC安装失败的问题(已解决) - 知乎 (zhihu.com)
后,感觉说的很吓人。。。但我是在cuda/lib的下面都没有找到libcusparse这个文件
后续重新创建了虚拟环境py3.9的进行安装
conda create -n PyG python=3.9
进入到环境之后安装pytorch2.0版本
官网下载的速度很慢,后续找了中国科学技术大学的源,参考Windows安装pytorch2.0.0(GPU)+pytorch-lightning2.0.0 - 知乎 (zhihu.com)
安装完后继续安装PyG
参考No module named torch_sparse, 及pytorch-geometric安装 - 知乎 (zhihu.com)
安装时要先安装四个包再安装torch-geometric,轮子如下
最后pip install torch-geometric即可