No module named ‘torch_geometric‘报错

目录

  • 一、打开当前环境所在的命令行
  • 二、查看torch版本和cuda版本
  • 三、下载安装所需要的组件
  • 四、安装pyg

一、打开当前环境所在的命令行

输入python 查看python版本
在这里插入图片描述

二、查看torch版本和cuda版本

import torch
torch__version__  #双下斜杠
torch.version.cuda

No module named ‘torch_geometric‘报错_第1张图片
可以看到 我的Python版本是3.6 torch版本是1.9.1 cuda版本是11.1

三、下载安装所需要的组件

https://pytorch-geometric.com/whl/
No module named ‘torch_geometric‘报错_第2张图片
cp后面的数字代表是自己的python版本
No module named ‘torch_geometric‘报错_第3张图片
将torch_cluster、torch_scatter、torch_sparse、torch_spline_conv都下下来,放到你刚刚打开python的命令行文件位置
No module named ‘torch_geometric‘报错_第4张图片
然后pip install 四个文件名.whl
No module named ‘torch_geometric‘报错_第5张图片
No module named ‘torch_geometric‘报错_第6张图片
No module named ‘torch_geometric‘报错_第7张图片
No module named ‘torch_geometric‘报错_第8张图片

四、安装pyg

pip install torch_geometric

No module named ‘torch_geometric‘报错_第9张图片
搞定

你可能感兴趣的:(GNN,pytorch,深度学习,python,graph)