pyg安装
pyg官网:pytorch_geometric documentation (pytorch-geometric.readthedocs.io)
win安装pyg各种报错,原因在于pytorch和pyg的版本问题,这里简单记录一下。
参考官方文档,前提已安装好pytorch,本人的配置为 torch1.11.0 torchvision0.12.0 cuda11.3 , 这里pyg安装为其对应版本。
pyg安装步骤如下(需要3个包,依次安装即可):
1. pip install torch-scatter -f https://data.pyg.org/whl/torch-1.11.0+cu113.html
2. 直接使用pip安装torch-sparse会报错,因此在链接https://data.pyg.org/whl/torch-1.11.0+cu113.html中下载whl版torch-sparse,离线安装即可
(离线安装whl教程可参考hl文件安装方法_寒幕小怪的博客-CSDN博客_whl文件怎么安装)
3. pip install torch-geometric
以上,即可安装成功!