这是没有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, profile
为import torch.autograd.profiler
,即
- 然后重新运行代码,没有报这个错误了,到此这个库没有问题了!!!
也有可能还会有其他问题,具体问题具体对待,欢迎大家一起交流 (^o^)/
注意收藏哦,防止下次找不到啦 (^o^)/
看见我这么努力的份上,点个赞再走吧!!