当前位置: 首页 > news >正文

正确安装 torch_geometric库

step1: 查看pytorch+cuda 版本

torch-scatter torch-sparse torch-cluster torch-spline-conv 这些关联包要与torch版本匹配。

import torch
print(torch.__version__)
print(torch.cuda.is_available())
torch.version.cuda

或者 pip list查看版本

step2: 下载对应版本的包文件

比如我的是1.13.0,cu117版本,那么我就装1.13.0 cu117版本对应的那四个关联包,如下命令:

对应版本下载链接:https://data.pyg.org/whl/

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html
pip install torch-geometric

step3: 注意torch-geometric的安装

若报错:AttributeError: module ‘torch_geometric.utils‘ has no attribute ‘scatter_‘
直接搜这个错误,没有找到相关的内容,大多数解决方案是对于pytorch出现的报错,应该是用这个库的人不多。在Github上torch-geometric库的讨论区中也没有找到相关的内容。抱着试一试的心态,我直接搜这个函数*torch_geometric.utils.scatter_*的相关内容。在一篇介绍torch-geometric安装的博文中,我找到了一句关于torch-geometric版本的说明,举例介绍了“torch_geometric.utils.scatter_这个方法,在1.5及以后的版本就不支持了,所以代码中有用到这个方法的,要不就另写一个函数代替,要不就把版本退到1.4.3”。于是我尝试把torch-geometric退回到1.4.3版本:

pip install torch-geometric==1.4.3

step4: 遇到的其他奇葩问题

OSError: Could not load shared object file: llvmlite.dll
pip 卸载: numba 和 llvmlite 包
然后: pip 安装

有人建议使用conda 安装,(慢且安装与其他的包冲突)本人未成功。改用pip 成功了!!!!

pip uninstall numba  llvmlite 
pip install numba  llvmlite 

step5: 参考

————————————————
参考链接:llvmlite.dll 错误解决

参考链接[1] torch-geometric 版本:https://blog.csdn.net/qq_40329272/article/details/111807624

参考链接[2] :https://blog.csdn.net/qq_46031119/article/details/128153390

http://www.lryc.cn/news/5826.html

相关文章:

  • 【Unity VR开发】结合VRTK4.0:自身移动(滑动)
  • G1垃圾回收器详解
  • tws耳机哪个牌子音质好?tws耳机音质排行榜
  • TIA博途中DB数据块清零的具体方法示例
  • iptables防火墙屏蔽指定ip的端口
  • JavaScript Math(算数) 对象
  • 超详细的JAVA高级进阶基础知识04
  • Python 运算符?
  • linux nuxt 部署 问题汇总
  • C++内存管理
  • 电子招投标系统源码之 —采购数字化转型快人一步,以大数据支撑供应链管理未来
  • ie获取cookie数据,中文乱码;cookie中文乱码终极解决办法
  • day16_关键字this和super丶就近原则和追根溯源原则
  • MySQL 共享锁 (lock in share mode),排他锁 (for update)
  • 类与对象(下)
  • feign技巧 - form方式传值
  • MATLAB | 情人节来绘制更立体的玫瑰花吧
  • 【Python表白代码】 2.14“Valentine‘s Day”“没别的意思 就是借着特殊日子说声喜欢你”你在哪儿?我去见你~(各种玫瑰源码合集)
  • 压力应变电桥信号隔离放大变送器差分输入0-±10mV/0-±20mV转0-20mA/0-10v
  • Linux系统之部署个人导航页
  • 四、Windows 平台安装 MongoDB
  • 浅谈应用安全测试工具
  • 四类(七种)排序算法总结
  • [oeasy]python0083_十进制数如何存入计算机_八卦纪事_BCD编码_Binary_Coded_Decimal
  • 理解框架的编译时与运行时
  • 推挽电路---采用二极管消除交越失真----克服交越失真的互补推挽输出电路图
  • day11_面向对象
  • 大数据处理学习笔记1.1 搭建Scala开发环境
  • VSCODE C++ 调用matplotlibcpp画图
  • 面对“开门红”,跨境支付如何寻求新增长曲线?