pytorch_geometric安装教程

最新推荐文章于 2025-04-23 17:31:19 发布

我是一只的猪

最新推荐文章于 2025-04-23 17:31:19 发布

阅读量2.2k

点赞数 1

文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/qq_40319989/article/details/113485508

版权

PyTorch-Geometric（PyG）是基于PyTorch的图数据处理库，适用于图学习算法。安装时需确保CUDA版本大于10.0，并根据PyTorch版本选择合适的库进行安装。在Linux环境下，可以从官方页面下载对应版本的whl文件，然后使用pip安装。

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

PyG 安装
PyG 全称是PyTorch-Geometric，是一个PyTorch基础上的一个库，专门用于图形式的数据，可以加速图学习算法的计算过程，比如稀疏化的图等。

先查看版本：

import torch;

print(torch.__version__)

print(torch.version.cuda)

: 在这里插入图片描述

注意 cuda版本必须是10.0以上

对于 PyTorch 版本为1.4.0，安装一下库：

其中${CUDA}取决于为本地的PyTorch安装环境，可以替换为：cpu, cu92, cu101 or cu102

#其中${CUDA}取决于为本地的PyTorch安装环境，可以替换为：cpu, cu92, cu101 or cu102
pip install torch-scatter==latest+${
   CUDA} -f https://pytorch-geometric.