Windows環境下PyTorch_geometric安裝踩坑

pytorch geometric在windows環境下安裝非常惡心，莫名其妙各種報錯。本帖針對GCC的編譯error提供解決方案。

---

一般報錯的包主要是兩個：

• torch-scatter
• torch-sparse

Building wheels for collected packages: torch-scatterBuilding 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: 33cp39-cp39-win_amd64cp39-abi3-win_amd64cp39-none-win_amd64cp38-abi3-win_amd64cp37-abi3-win_amd64cp36-abi3-win_amd64cp35-abi3-win_amd64cp34-abi3-win_amd64cp33-abi3-win_amd64cp32-abi3-win_amd64py39-none-win_amd64py3-none-win_amd64py38-none-win_amd64py37-none-win_amd64py36-none-win_amd64py35-none-win_amd64py34-none-win_amd64py33-none-win_amd64py32-none-win_amd64py31-none-win_amd64py30-none-win_amd64cp39-none-anypy39-none-anypy3-none-anypy38-none-anypy37-none-anypy36-none-anypy35-none-anypy34-none-anypy33-none-anypy32-none-anypy31-none-anypy30-none-any

對照tag做修改，改完之后的文件名應該如下。

torch_scatter-2.0.6-cp39-cp39-win_amd64.whl

然后就可以用pip安裝了。

總結

以上是生活随笔為你收集整理的Windows环境下PyTorch_geometric安装踩坑的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。