图神经网络torch_geometric库的使用

发布于：2022-12-16 ⋅ 阅读:(302) ⋅ 点赞:(0)

1.torch_scatter/torch-sparse的安装

千里之行，始于足下。

先从安装两个依赖库开始。

直接使用pip install ***会报错，然后到官方页面寻找安装方式，使用形如：

pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.12.0+${CUDA}.html

方式安装，仍然不行。

参考blog的安装方式，成功安装。

整个过程分别为：

1）确认当前安装的torch的版本

import
torch.__version__
'1.8.1+cu111'

2）到官网找到对应的链接，
在这里插入图片描述
点进去查找对应的wheel，我这里使用的windows平台python3.7版本，

分别下载对应的whl；

3）使用本地方式进行安装：

pip install E:\torch_sparse-0.6.9-cp37-cp37m-win_amd64.whl
pip install E:\torch_scatter-2.0.8-cp37-cp37m-win_amd64.whl

就可以成功安装了。