一、说明
安装torch-geometric需要先安装torch_cluster、torch_scatter、torch_sparse、torch_spline_conv四个文件
(ps: 安装教程主要目的是作为自己的备忘录,所有方法都具有时效性,随着时间流逝失效的风险会逐渐增大,如若没用请见谅)
二、安装
进入https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html
注意:网址中torch-*.**.*+cu***对应的是自己环境中torch版本,我的是1.12.1+cu113,所以网址如此。(这里安装的是cuda版本,cpu版本的可能需要自己再找一下其他博客)
进入之后,为下面界面
界面中,紫色的框直接忽略,红色框中就是需要下载的文件(框住的是torch_cluster,另外三个同理下载)。其中请注意自己的python版本和系统类别,下载合适的文件。
win系统 + python3.9 + torch1.12.1+cu113版本的下载内容如下:
下载完成后直接pip安装至需要的虚拟环境中。
四个文件全部安装完成后,执行
pip install torch-geometric
完工。
验证如下: