1. torch_geometric 环境搭建
- No module named torch_sparse, 及pytorch-geometric安装
- pytorch-geometric-whl
- 图神经网络库PyTorch geometric(PYG)零基础上手教程
找到自己对应的版本即可
我的 CUDA版本 : 10.2
Pytorch版本 : 1.10.0
- 根据具体情况选择版本
![image-20220304171143972](https://haloos.oss-cn-beijing.aliyuncs.com/typero/image-20220304171143972.png)
pip install torch-scatter==2.0.9 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-sparse==0.6.12 -f https://pytorch-geometric.com/whl/torch-1.10.0+cu102.html
pip install torch-geometric