pyg也就是pytorch geometri。不可以直接使用pip下载,需要下载whl依赖。
自行补充whl
github官网链接
whl补充地址
需要补充三个,torch-scatter、torch-sparse和torch-cluster请自己对应版本
注意要python3.7到3.11的才能使用 pyg
查看版本代码
import torch
print(torch.__version__) #pytorch版本
print(torch.version.cuda) #cuda版本
将whl依赖下载在本地,使用pip下载依赖,最后使用pip install torch-geometric即可
或者去官网 官网
自行调整配置,先执行底下的额外依赖
如果一直三个依赖都已经下载了,还是不能安装,尝试切换其他镜像网站如下所示:
pip install torch-geometric -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com //豆瓣