torch版本对应的torch_geometric与torch-sprse/cluster/scatter库的正确安装
使用上述标红命令即可快速安装需要的包(确定自己环境中安装的pytorch版本以及cuda版本,使用对应的命令即可)
文章共246字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
torch_geometric官网:
Installation — pytorch_geometric documentation
使用上述标红命令即可快速安装需要的包(确定自己环境中安装的pytorch版本以及cuda版本,使用对应的命令即可)
如安装的pytorch为1.60,cuda为10.1,则:
pip install torch-scatter torch-sparse torch-cluster -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html 使用以上命令安装相关包即可。
关于核心的torch_geometric包,选择版本之后能看到右边的Quick Star信息,表示其最适合兼容的pytorch版本。但也不是唯一,如我的环境中的相关版本如下,可以正常运行:
更多推荐
已为社区贡献14条内容
所有评论(0)