【冰糖Python】PyTorch Geometric 安装

最新推荐文章于 2025-01-15 12:54:01 发布

冰糖不在家

最新推荐文章于 2025-01-15 12:54:01 发布

阅读量586

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 图神经网络文章标签： python

本文链接：https://blog.csdn.net/xiaoxiao_ziteng/article/details/114022405

本文档详细介绍了如何通过二进制文件安装PyTorchGeometric。首先确认已安装PyTorch及其对应的CUDA版本，然后根据版本安装torch-scatter、torch-sparse、torch-cluster、torch-spline-conv等依赖库，最后安装torch-geometric。提供了针对PyTorch1.7.1CPU版本的安装命令示例，并给出了查看最新版本的链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

此文记录通过二进制文件安装 PyTorch Geometric 的过程

1）确认 PyTorch 是否已经安装，可参考此文安装 PyTorch，比如安装了 PyTorch 1.7.1 cpu 版本

2）确认所安装的 PyTorch 对应的 CUDA 的版本，比如 CUDA 11.0 或 cpu 版本

3）根据 1）和 2）对应的版本安装以下依赖库

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

其中，${TORCH} 和 ${CUDA} 分别替换为 1）和 2）中确认的版本

${TORCH} 可选 {1.4.0, 1.5.0, 1.6.0, 1.7.0}，PyTorch 1.7.1 选 1.7.0

${CUDA} 可选 {cpu，cu92，cu101，cu102，cu110}

则假设使用 PyTorch 1.7.1 cpu 版本，安装以上依赖为：