PyTorch Geometric详细安装步骤

文章详细介绍了如何安装PyTorchGeometric(PyG),首先查看PyTorch和CUDA版本(1.11.0+cu115),然后通过pip安装相应的依赖文件,如torch_cluster,torch_scatter和torch_spline_conv,最后安装PyG本身。这是一个针对Python3.9和WindowsAMD64系统的安装过程。
摘要由CSDN通过智能技术生成

PyTorch Geometric安装

PyG是当前最流行和广泛使用的GNN库。
PyG官网
进入官网,点击install,选择如下的方式进行安装:
在这里插入图片描述

进入here安装依赖:
先查看pytorch版本和cuda版本:

import torch
print(torch.__version__)
1.11.0+cu115
print(torch.version.cuda)
11.5

选择对应版本的依赖文件:
在这里插入图片描述
进入对应的链接选择正确的python版本和操作系统的依赖:
在这里插入图片描述
利用pip进行下载:

pip insatll https://data.pyg.org/whl/torch-1.11.0%2Bcu115/torch_cluster-1.6.0-cp39-cp39-win_amd64.whl
pip install https://data.pyg.org/whl/torch-1.11.0%2Bcu115/torch_scatter-2.0.9-cp39-cp39-win_amd64.whl
pip install https://data.pyg.org/whl/torch-1.11.0%2Bcu115/torch_spline_conv-1.2.1-cp39-cp39-win_amd64.whl

依赖下载完之后,安装PyG

 pip install torch-geometric
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值