代码运行出现:No module named ‘torch_geometric‘

这是没有torch_geometric库导致的,但是不能像一般库安装‘pip install 库名’或者‘conda install 库名’进行安装,经常会报错!!!

需要先安装四个小部件再安装torch_geometric,具体安装步骤如下:

step 1:查看自己环境下安装的pytorch和cuda版本

以本人的为例,我的pytorch是1.6.0,cuda是10.1
在这里插入图片描述

step 2:进入相应版本下载四个小部件whl文件

点击进入链接https://pytorch-geometric.com/whl/,如下所示。
在这里插入图片描述
按照本人第一步查的信息,选择其中‘torch-1.6.0+cu101’进入:
在这里插入图片描述
选择适合你的进行四个小部件下载,我的环境python是3.8的,本人都选择的是最高版本,如下四个:
在这里插入图片描述

step 3:按照如下顺序安装四个小部件

pip install torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
pip install torch_scatter-2.0.6-cp38-cp38-linux_x86_64.whl
pip install torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl
pip install torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl

显示如下即安装成功!
在这里插入图片描述

step 4:安装torch_geometric

pip install torch_geometric

输出:
在这里插入图片描述安装成功!!!

本文解决办法借鉴博客:https://blog.csdn.net/weixin_47779152/article/details/120570367

运行代码依旧报错"No module named 'torch.profiler"

上述安装成功,运行代码发现报如下错误:
在这里插入图片描述
查了资料,其中普遍讲是pytorch版本低了,需要提高版本!本人也查了,好像是只有torch1.10以上的版本才有torch.profiler这个库,但是我又不想升级,很多库是根本对应版本安装的,我担心升级pytorch版本后其他的报错!!继续找材料,最后发现这个网站有个解决办法:https://www.cnblogs.com/dive-into/p/16953369.html

具体步骤:

  • 按照报错路径找到profile.py文件
  • 修改from torch.profiler import ProfilerActivity, profileimport torch.autograd.profiler,即
    在这里插入图片描述
  • 然后重新运行代码,没有报这个错误了,到此这个库没有问题了!!!

也有可能还会有其他问题,具体问题具体对待,欢迎大家一起交流 (^o^)/

注意收藏哦,防止下次找不到啦 (^o^)/
看见我这么努力的份上,点个赞再走吧!!

猜你喜欢

转载自blog.csdn.net/weixin_43994864/article/details/132292340