怎么在服务器(Linux)上安装PYG

一.检查torch版本信息

复制下面代码段查看当前python版本信息和pytorch版本:
运行代码:
import sys
import torch
print(f"Python 版本: {sys.version}")
print(f"PyTorch 版本: {torch.__version__}")
输出:
Python 版本: 3.8.2rc2 
PyTorch 版本: 2.3.1+cu121
得记住这些信息,后面需要使用

二.下载关键包

单纯使用
pip install pip install torch_geometric
会漏包,导致运行出错。
比如说使用knn_graph生成边的时候会报错

因此需要到下方链接上下载下面图中的四个包
链接:pyg补充包
在这里插入图片描述
这里需要根据自己的python版本和cuda版本选择包
若我们版本信息为:

输出:
Python 版本: 3.8.2rc2 
PyTorch 版本: 2.3.1+cu121

我们选择:
在这里插入图片描述

在这个链接里面选择对应的python版本号和系统型号,这里我们的python版本是3.8。因此我们选择3.8版本:(由于我们要装在服务器Linux版本这里选择linux版本,如果电脑是其他系统选择响应的版本)

在这里插入图片描述下载完包后将包放到项目文件下面(一定要激活自己的环境再开始第三章)

三.命令行安装补充包和pyg

pip install torch_cluster-1.6.3+pt23cu121-cp38-cp38-linux_x86_64.whl
pip install torch_scatter-2.1.2+pt23cu121-cp38-cp38-linux_x86_64
pip install torch_sparse-0.6.18+pt23cu121-cp38-cp38-linux_x86_64.whl
pip install torch_spline_conv-1.2.2+pt23cu121-cp38-cp38-linux_x86_64.whl
pip install torch_geometric

这样基本上就大功告成了
要点:

  1. 安装补充包
  2. 版本对应:python版本、torch版本、cuda版本、系统版本
  • 8
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

代码飞速跑

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值