ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

Windows环境下PyTorch_geometric安装踩坑

2021-05-22 15:01:08  阅读:741  来源: 互联网

标签:none amd64 Windows win PyTorch cp39 abi3 geometric any


Windows环境下PyTorch_geometric安装踩坑

pytorch geometric在windows环境下安装非常恶心,莫名其妙各种报错。本帖针对GCC的编译error提供解决方案。


一般报错的包主要是两个:
  • torch-scatter
  • torch-sparse
Building wheels for collected packages: torch-scatter
  Building wheel for torch-scatter (setup.py) ... error

根据Github上的#226这是使用GCC版本不对造成的。
根据PyTorch geometric官方FAQ FAQ上的描述

return _and<is_constructible<_Elements, _UElements&&>…>::value: Ensure that your gcc version is at least 4.9 (and below 6) (see Issue#25b). You will also need to reinstall PyTorch because gcc versions must be consistent across all PyTorch packages.

GCC的版本不能低于4.9并且要低于6(真麻烦。。。)。


解决方案:

使用mingw直接安装会安装最新版6+版本。找了一圈发现mingw也提供older versions download pages
直接拉到底能够找到5.4.0版本,下载解压,然后配置环境变量就可以了。
在这里插入图片描述
成功之后应该会显示
在这里插入图片描述


接下来就是下载pytorch geometric的相关whl文件。
下载链接pytorch-geometric
选择自己系统对应的cuda和pytorch版本
我是cuda11.1和torch1.8.1

接下来有一个小坑
下载好的文件应该长这样

torch_scatter-2.0.6-cp38-cp38-win_amd64.whl

然而我的python是3.9的版本直接pip install是装不上的。先要查看下适配的标签文件。

 pip debug --verbose

重点是Compatible tags

Compatible tags: 33
  cp39-cp39-win_amd64
  cp39-abi3-win_amd64
  cp39-none-win_amd64
  cp38-abi3-win_amd64
  cp37-abi3-win_amd64
  cp36-abi3-win_amd64
  cp35-abi3-win_amd64
  cp34-abi3-win_amd64
  cp33-abi3-win_amd64
  cp32-abi3-win_amd64
  py39-none-win_amd64
  py3-none-win_amd64
  py38-none-win_amd64
  py37-none-win_amd64
  py36-none-win_amd64
  py35-none-win_amd64
  py34-none-win_amd64
  py33-none-win_amd64
  py32-none-win_amd64
  py31-none-win_amd64
  py30-none-win_amd64
  cp39-none-any
  py39-none-any
  py3-none-any
  py38-none-any
  py37-none-any
  py36-none-any
  py35-none-any
  py34-none-any
  py33-none-any
  py32-none-any
  py31-none-any
  py30-none-any

对照tag做修改,改完之后的文件名应该如下。

torch_scatter-2.0.6-cp39-cp39-win_amd64.whl

然后就可以用pip安装了。

标签:none,amd64,Windows,win,PyTorch,cp39,abi3,geometric,any
来源: https://blog.csdn.net/WildCatFish/article/details/117159123

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有