torch_geometric的安装过程和Error building wheel for torch-scatter报错的解决

我一开始去网上搜教程来安装,结果报错,搞了很久。后来发现其实官方文档中写得很清楚了,接下来的安装过程其实就是参照官方文档的Installation via Pip Wheels~同时记录了自己遇到的问题。

官方文档地址:Installation — pytorch_geometric documentation

第一步:确认pytorch和cuda版本

确认pytorch版本:

python -c "import torch; print(torch.__version__)"

确认安装pytorch时选择的cuda版本:

python -c "import torch; print(torch.version.cuda)"

如果pytorch版本符合Quick Start里的要求,直接按照Quick Start安装应该就行,不行的话再往下看试试。

我的pytorch版本是1.9.1,cuda版本是11.1

官方文档里列出的第二种方式是用Anaconda安装,说pytorch版本大于1.8.0即可。但是我试了一下anaconda,报错说有冲突:Found conflicts! Looking for incompatible packages. 等半天也装不好,就直接ctrl+c

第二步:通过Pip Wheels安装

官方文档中给出的方法:

pip install pyg-lib torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

${TORCH}表示pytorch版本号,${CUDA}表示cuda版本号。比如pytorch版本是1.13.*,cuda版本是11.6,就写作:

pip install pyg-lib torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.13.0+cu116.html
pip install torch-geometric

包括搜到的一些方法也说用上述命令就能行,但我不行!一直在build wheel时报错,比如安装torch-scatter显示Error building wheel for torch-scatter和一大串错误信息。

解决方法是手动下载whl文件,再安装。

官方文档上给出了支持的pytorch和cuda版本:https://data.pyg.org/whl/
在这里找到自己版本对应的,比如我就是torch-1.9.1+cu111
然后找到每个包对应python版本和系统型号下载。根据官方文档,torch-cluster和torch-spline-conv是可选的,其他包都要装一下。

下载后,进入放.whl文件的目录下,执行

pip install torch_scatter-2.0.9-cp38-cp38-win_amd64.whl

每个下载好的whl文件安装好后,执行

pip install torch_geometric

就OK啦!

参考资料:

How to install whl files in Anaconda - deparkes

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值