看论文代码的时候发现需要一个叫torch_cluster的包,本来以为就是那种随便pip一下就可以下载下来的包,但最终还是被折磨了一天才发现这是pyg(pytorch_geometric)的一个子包,然后正确的安装上了。在csdn上找了很多方法都比较容易出错,至少我是没有安装成功。最终安装方法是从B战的一个视频上找到的。这里做一下图文的转述,这样习惯在csdn上的找的人可以少很多弯路。
1,确认自己的pytorch版本和cuda的版本。如果使用的是cpu版本的pytorch就不用看cuda的版本了。
运行这样一个代码。
import torch print(torch.__version__)
结果:
1.12.1就是pytorch的版本,cu116代表cuda是11.6版本。
2,使用GitHub安装
网址为https://github.com/pyg-team/pytorch_geometric
这是pyg的GitHub上的代码
向下拉有各种安装方式的介绍。不要用conda或在线安装,很容易出错。一定要先下载whl文件再使用pip来进行本地安装。
找到这里
点击here
pyg主要有四个子包
torch-cluster
torch-scatter
torch-sparse
torch-spline-conv
分别安装好这四个包之后再安装pyg本体。
进去后的界面先根据第一步里面查到的pytorch版本和cuda版本来选择。
这里我就选torch-1.12.1+cu116
再进去就到四个子包的whl文件下载了
cp是python版本的意思,这里根据python版本和操作系统分别依次下载四个子包。
之后在你需要安装pyg的虚拟环境里面使用 pip install + 文件名.whl 来安装四个子包包。这里可能会有路径的一些问题,不过很简单。
最后再运行一下
pip install torch-geometric
即可。