python安装torch-geometric库

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

上面是别人提供的安装方法,这里${CUDA} 和${TORCH} 是你的torch和相应的CUDA版本,比如我的torch版本是1.8.0 CUDA版本是11.1,安装torch-scatter的指令就是

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html

但是按照上述方法操作会出现问题,报错一般如下,

总之就是一串红,大概意思是说你的gcc版本不对(千万不要被这个误导!!!)

网上提供的解决方案要么毫无作用,要么就是安装后import时又会报错,

虽然并未完全解决我的问题,但却为问题解决提供了一些思路(版本依赖问题)

其实可以先看一下安装指令 -f 后面的链接(我的torch及cuda版本对应的链接即https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html)里到底有什么,打开后发现是一些轮子资源

于是我发现这样一个问题,我安装的torch-scatter默认版本是2.0.9,如下图(因为我下载过了所以有cache),但在上面的链接中发现并没有对应的版本,于是我想可能是这里出了问题

 将torch-scatter的版本改为上面的链接中包含的版本(2.0.6~2.0.8),这里我改成2.0.8,再次pip install,安装成功!(如下图)

 按这个思路依次安装其他几个包,最后再安装torch-geometric,我执行的指令依次是:

pip install torch-scatter==2.0.8 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-sparse==0.6.9 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-geometric

 均安装成功,最后试一下import,正常!

(如有问题,欢迎指正~)

  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值