win10下安装pyg出现torch_scatter报错解决

  • 时间:
  • 来源:互联网
  • 文章标签:

根据官方文档,按照在线安装教程,总是或多或少出现各种bug,让人很是着急,刚开始搜资料也没找到什么方案,感觉尝试后依旧无效,自我放弃后,现在找到万能解决方案!

那就是离线安装吧!绝对安装不会出问题,但是一定要注意torch cuda torch_scatter等相关包版本的对应关系,拿torch1.6举栗子

第一步:根据torch 和cuda对应关系安装(本人安装的是torch1.6 +cuda10.2)下面两个文件(下载位置)

第二步:基于第一步,加上自己安装的python版本,安装下面4个文件(下载位置)

本人创建的虚拟环境python版本为3.6,因此选择cp36,下载时直接选择torch1.6+cu102,另外torch1.6是只有cpu的情况

第三步:安装最后一个 pip install torch-geometric,安装成功后如下

最后:进行验证

以上详细安装步骤参考本人另一篇:pyg离线安装

 

本文链接http://www.taodudu.cc/news/show-1944670.html