提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


前言

提示:记录一下安装配置几个依赖库直接的关系:

踩坑:标题的那几个库,直接安装会有很大概率导致版本不兼容造成报错,本文用于记录自己的踩坑解决方案。


提示:以下是本篇文章正文内容,下面解决方案可供参考

一、查看自己的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

安装成功!

总结

跟着教程一点点来就可以了。如果有不懂或者书写有误的地方,欢迎指正,可以私信,或者加入我们的微信公众号。

在这里插入图片描述

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐