安装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

 这是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

 即可。

  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值