【关于pytorch-geometric、torch_cluster、torch_scatter、torch_sparse、torch_spline的版本对应及安装】
踩坑:标题的那几个库,直接安装会有很大概率导致版本不兼容造成报错,本文用于记录自己的踩坑解决方案。提示:以下是本篇文章正文内容,下面解决方案可供参考跟着教程一点点来就可以了。如果有不懂或者书写有误的地方,欢迎指正,可以私信,或者加入我们的微信公众号。
·
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
提示:记录一下安装配置几个依赖库直接的关系:
踩坑:标题的那几个库,直接安装会有很大概率导致版本不兼容造成报错,本文用于记录自己的踩坑解决方案。
提示:以下是本篇文章正文内容,下面解决方案可供参考
一、查看自己的torch版本?
安装之前,先查看自己的torch版本,查看torch版本的方法很多,这里我使用:
pip list
直接列出所有的第三方库,寻找torch
所以这里的版本就是2.4.1其中 cuda版本是12.4
二、安装依赖库
1.寻找对应版本
页面效果如下:
这里解释一下,示例:假设你的torch版本是2.1.2 并且 cuda 版本是 12.1 你就点击 torch-2.1.2+cu121 接着会跳转到:
分为五个板块,这里接着选择自己对应的版本,下载即可。其中解释一下,
torch_cluster-1.6.2+pt21cu121-cp310-cp310-win_amd64.whl 表明的是 pt21cu121 torch2.1版本并且cu12.1 cp310 python3.10版本
win_amd64 你要安装在win平台。 自己根据自己的需要安装即可。
下载到本地之后,文件是这样的:
2.安装
安装如下(示例):
pip install pyg_lib-0.4.0+pt24cu124-cp310-cp310-win_amd64.whl
pip install torch_sparse-0.6.18+pt24cu124-cp310-cp310-win_amd64.whl
pip install torch_cluster-1.6.3+pt24cu124-cp310-cp310-win_amd64.whl
pip install torch_scatter-2.1.2+pt24cu124-cp310-cp310-win_amd64.whl
pip install torch_spline_conv-1.2.2+pt24cu124-cp310-cp310-win_amd64.whl
这些全都要安装。
安装成功!
pip install pytorch-geometric
安装成功!
总结
跟着教程一点点来就可以了。如果有不懂或者书写有误的地方,欢迎指正,可以私信,或者加入我们的微信公众号。
更多推荐
所有评论(0)