torch-geometric的安装,pip手动安装4个依赖包

torch-geometric一直装不上,因为四个依赖包一直装不上(torch_cluster、torch_scatter、torch_sparse、torch_spline_conv),网上看了好多教程,一直都是各种fail bulid wheel

最后还是自己手动下载whl文件,然后使用pip install + whl路径成功安装,下面是方法:

0.如果之前失败安装了4个依赖包,那么先卸载掉

pip uninstall torch-scatter torch-sparse torch-cluster torch-spline-conv

1.查看自己的pytorch和cuda版本,例如torch-1.10.0+cu113

2.打开pyg官方依赖下载页data.pyg.org/whl/

3.找到自己版本的torch和cuda,例如torch-1.10.0+cu113

4.点开以后,分别下载四个依赖包,这里选择自己的python版本以及操作系统,我这里以python3.7+windows为例。

注:这里可能会有多个版本,比如cluster就有1.6.0和1.5.9,我是选了更新一点的那个,应该两个版本的都是可以用的

5.安装依赖包,使用pip install + whl文件,分别安装四个依赖包

pip install C:\Users\Administrator\Desktop\4whl\torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl

pip install C:\Users\Administrator\Desktop\4whl\torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl

pip install C:\Users\Administrator\Desktop\4whl\torch_sparse-0.6.13-cp37-cp37m-win_amd64.whl

pip install C:\Users\Administrator\Desktop\4whl\torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl

6.最后,安装pyg

pip install torch-geometric==x.x.x

7.安装完毕,我们import测试一下:

python

import torch_geometric

exit()

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值