Torch_geometric安装记录
buaaty
2023年10月31日 10:41
收录于文集
共1篇

最近学习GNN,需要安装Torch_geometric库,安装了许久,终于成功了,简单记录一下安装过程和踩的坑,以防后续需要

首先通过Anaconda prompt重新安装一个虚拟环境,或者使用之前的虚拟环境,这里我使用的是之前已有的虚拟环境,版本分别为:

Python == 3.7.1;

Torch == 1.5.0;

之前也尝试新建了个虚拟环境,但是import torch总是出错,所以用了已有的正确的虚拟环境进行安装torch-geometric库。

然后在官网上分别下载以下四个文件(CPU,无GPU):

然后按照https://zhuanlan.zhihu.com/p/602615941?utm_id=0 说法,依次使用pip install安装torch_scatter、torch_sparse、torch_cluster和torch_spline_conv(打乱顺序安装可能会出现文件缺失的问题)。

最后 安装:pip install torch-geometric。

需要注意的是如果直接使用pip install torch-geometric命令进行安装,会安装最新版本,进而可能会出现程序报错,如:ModuleNotFoundError: No module named 'torch.profiler'

然后我按照https://zhuanlan.zhihu.com/p/659534878对profile.py程序进行修改,但会出现其他问题,后来根据https://blog.csdn.net/m0_55684014/article/details/133578645指导,尝试将torch-geometric进行版本降级,即

先卸载已安装的库“pip uninstall torch_geometric”

然后指定版本“pip install torch_geometric==2.0.4”

这时在python环境下运行“import torch” 和“import torch_geometric”均正确,

且通过https://zhuanlan.zhihu.com/p/602615941?utm_id=0里的测试案例,在pycharm里运行得到正确如下输出结果

在pycharm运行时,要记得选择在这个安装了torch_geometric库的python环境下运行

至此,完成了gnn所需的torch_geometric安装!