Windows環(huán)境下PyTorch_geometric安裝踩坑

pytorch geometric在windows環(huán)境下安裝非常惡心，莫名其妙各種報(bào)錯(cuò)。本帖針對(duì)GCC的編譯error提供解決方案。

---

一般報(bào)錯(cuò)的包主要是兩個(gè)：

• torch-scatter
• torch-sparse

Building wheels for collected packages: torch-scatterBuilding wheel for torch-scatter (setup.py) ... error

---

根據(jù)Github上的#226這是使用GCC版本不對(duì)造成的。
根據(jù)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直接安裝會(huì)安裝最新版6+版本。找了一圈發(fā)現(xiàn)mingw也提供older versions download pages
直接拉到底能夠找到5.4.0版本，下載解壓，然后配置環(huán)境變量就可以了。

成功之后應(yīng)該會(huì)顯示

---

接下來就是下載pytorch geometric的相關(guān)whl文件。
下載鏈接pytorch-geometric
選擇自己系統(tǒng)對(duì)應(yīng)的cuda和pytorch版本
我是cuda11.1和torch1.8.1

接下來有一個(gè)小坑
下載好的文件應(yīng)該長(zhǎng)這樣

torch_scatter-2.0.6-cp38-cp38-win_amd64.whl

然而我的python是3.9的版本直接pip install是裝不上的。先要查看下適配的標(biāo)簽文件。

pip debug --verbose

重點(diǎn)是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

對(duì)照tag做修改，改完之后的文件名應(yīng)該如下。

torch_scatter-2.0.6-cp39-cp39-win_amd64.whl

然后就可以用pip安裝了。

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。