安装pytorch_geometric

        看论文代码的时候发现需要一个叫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

安装pytorch_geometric_第1张图片

 这是pyg的GitHub上的代码

向下拉有各种安装方式的介绍。不要用conda或在线安装,很容易出错。一定要先下载whl文件再使用pip来进行本地安装。

找到这里安装pytorch_geometric_第2张图片

点击here

pyg主要有四个子包

torch-cluster
torch-scatter
torch-sparse
torch-spline-conv

分别安装好这四个包之后再安装pyg本体。

安装pytorch_geometric_第3张图片

 进去后的界面先根据第一步里面查到的pytorch版本和cuda版本来选择。

这里我就选torch-1.12.1+cu116

再进去就到四个子包的whl文件下载了

安装pytorch_geometric_第4张图片

cp是python版本的意思,这里根据python版本和操作系统分别依次下载四个子包。

之后在你需要安装pyg的虚拟环境里面使用 pip install + 文件名.whl 来安装四个子包包。这里可能会有路径的一些问题,不过很简单。

最后再运行一下

pip install torch-geometric

 即可。

你可能感兴趣的:(pip,python)