如何在服务器(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
这样基本上就大功告成了
要点:
- 安装补充包
- 版本对应:python版本、torch版本、cuda版本、系统版本