【解决】安装torch_geometric的一些坑

最近在考虑用GNN,于是首先安装torch_geometric库。。。遇到一些小坑。。。供大家参考

一、查看环境

  1. 打开虚拟环境
conda activate work
  1. 查看torch和cuda版本
import torch
print(torch.__version__)

我这里的版本是1.10+11.1
3. 找对应版本的torch_geometric进行安装,注意这里用wheel文件安装,在安装geometric之前需要先按顺序安装如下内容:(点here下载必要依赖)
第一步安装pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
第二步安装pip install torch_sparse-0.6.13-cp37-cp37m-win_amd64.whl
第三步安装pip install torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl
第四步安装pip install torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl
第五步安装pip install torch_geometric
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

  1. 在pycharm中输入import torch_geometric,没有显示报错即为安装成功。

报错解决

报错一:DLL load failed: 找不到指定的程序。

和我之前安装sklearn时候报的错一毛一样!
就不该动numpy
numpy是用wheel文件安装的,但是这次提示我升级,我直接用的pip install,导致出错(据说是由于在线pip install numpy会没有mkl),于是又找到wheel文件卸载numpy重新安装numpy,报错解决!

报错二

在这里插入图片描述
原因是因为PYG 2.3 requires that at least Pytorch 1.12 is installed.
解决方案
直接升级torch或者降级geometric就好了

后续写:介绍geometric
官方API:https://pytorch-geometric.readthedocs.io/en/latest/

  • 17
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值