从零开始学习神经网络 - torch_geometric库下载
·
1. torch_geometric 环境搭建
- No module named torch_sparse, 及pytorch-geometric安装
- pytorch-geometric-whl
- 图神经网络库PyTorch geometric(PYG)零基础上手教程
找到自己对应的版本即可
我的 CUDA版本 : 10.2
Pytorch版本 : 1.10.0

- 根据具体情况选择版本
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
更多推荐
所有评论(0)