【PyG】windows下安装PyTorch Geometric

最新推荐文章于 2024-06-02 12:49:05 发布

???873

最新推荐文章于 2024-06-02 12:49:05 发布

阅读量901

点赞数 1

分类专栏： Python 文章标签： python 深度学习 pytorch 神经网络计算机视觉

本文链接：https://blog.csdn.net/qq_17047265/article/details/106055708

版权

在Windows系统中，使用Python 3.6.8和CPU环境安装PyTorch Geometric时遇到了一系列问题，如torch_sparse模块缺失、scatter等包安装失败以及编译错误。尝试了各种方法，包括手动安装torch-scatter、torch-sparse等依赖，但问题依然存在。最终解决办法是确定各依赖包的匹配版本，如torch-cluster 1.2.4、torch-geometric 1.2.1、torch-scatter 1.1.2、torch-sparse 0.4.0和torch-spline-conv 1.0.6。成功安装后，可以应用于图神经网络的卷积层搭建。

摘要由CSDN通过智能技术生成

安装:

torch-geometric== 1.2.1
（最新版为1.4.3，注意各依赖包版本适配）

配置：

win10, cpu, python3.6.8, 无conda
torch≥1.1.0

遇到问题：

1.from torch_sparse import coalesce
ImportError: No module named ‘torch_sparse’

2.cannot install scatter, sparse, cluster packages

3.error: command ‘D:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\BIN\x86_amd64\cl.exe’ failed with exit status 2

4.信息: 用提供的模式无法找到文件

5.error: option --cpu not recognized

试过但没有解决问题的方法

pip install --verbose --no-cache-dir torch-scatter
pip install --verbose --no-cache-dir torch-sparse
pip install --verbose --no-cache-dir torch-cluster
pip install --verbose --no-cache-dir torch-spline-conv
pip install torch-geometric