图神经网络torch_geometric库的使用

图神经网络torch_geometric库的使用

  • 1.torch_scatter/torch-sparse的安装
  • 参考文献

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)到官网找到对应的链接,
图神经网络torch_geometric库的使用_第1张图片
点进去查找对应的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

就可以成功安装了。

参考文献

[1] https://pypi.org/project/torch-sparse/
[2] pytorch使用之torch_sparse安装

你可能感兴趣的:(深度学习,神经网络,pytorch,深度学习)