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

以上,即可安装成功!

 

posted @ 2022-08-30 20:36  vv_869  阅读(349)  评论(0编辑  收藏  举报