• 首页 首页 icon
  • 工具库 工具库 icon
    • IP查询 IP查询 icon
  • 内容库 内容库 icon
    • 快讯库 快讯库 icon
    • 精品库 精品库 icon
    • 问答库 问答库 icon
  • 更多 更多 icon
    • 服务条款 服务条款 icon

torch相关版本对应依赖23年最新

武飞扬头像
大渡水
帮助1

torch、torch-scatter、torch-sparse版本依赖问题

在运行程序的时候出现“ModuleNotFoundError: No module named 'torch_scatter' ”问题

  1. 不能直接下载,容易出现依赖不匹配。需要下载相应的依赖版本,首先查询cuda版本

linux命令行中(在你自己的虚拟环境下)输入nvcc -V,查看对应cuda版本

学新通

我的cuda版本是11.6,如果没有下载torch库,就访问官网进行下载需要的版本:Previous PyTorch Versions | PyTorch

学新通

(由于我已经下载了torch库,torch==1.13.1)

学新通
  1. 下载对应版本的torch_scatter

打开https://pytorch-geometric.com/whl/

可以看到很多torch和cuda组合的版本,找到对应torch以及cuda版本

学新通

torch-1.13.1 cu116版本下有相应的torch_cluster、torch_scatter、torch_sparse、torch_spline_conv四个与之版本依赖的库,pt113代表对应的torch版本,cp37代表所对应的python版本为3.7,有linux和windows版本

学新通
  1. 安装.whl文件

先切换到下载的whl文件所在目录,在命令行输入:

 pip install torch_scatter-2.1.0 pt113cu116-cp37-cp37m-linux_x86_64.whl
学新通

至此,恭喜你,解决了一个问题

这篇好文章是转载于:学新通技术网

  • 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
  • 本站站名: 学新通技术网
  • 本文地址: /boutique/detail/tanhgcgabe
系列文章
更多 icon
同类精品
更多 icon
继续加载