torch_geometric安装
我一开始去网上搜教程来安装,结果报错,搞了很久。后来发现其实官方文档中写得很清楚了,接下来的安装过程其实就是参照官方文档的Installation via Pip Wheels
~同时记录了自己遇到的问题。
官方文档地址:Installation — pytorch_geometric documentation
第一步:确认pytorch和cuda版本
确认pytorch版本:
python -c "import torch; print(torch.__version__)"
确认安装pytorch时选择的cuda版本:
python -c "import torch; print(torch.version.cuda)"
如果pytorch版本符合Quick Start
里的要求,直接按照Quick Start
安装应该就行,不行的话再往下看试试。
我的pytorch版本是1.9.1,cuda版本是11.1
官方文档里列出的第二种方式是用Anaconda安装,说pytorch版本大于1.8.0即可。但是我试了一下anaconda,报错说有冲突:Found conflicts! Looking for incompatible packages.
等半天也装不好,就直接ctrl+c了
第二步:通过Pip Wheels安装
官方文档中给出的方法:
pip install pyg-lib torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric
${TORCH}
表示pytorch版本号,${CUDA}
表示cuda版本号。比如pytorch版本是1.13.*,cuda版本是11.6,就写作:
pip install pyg-lib torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.13.0+cu116.html
pip install torch-geometric
包括搜到的一些方法也说用上述命令就能行,但我不行!一直在build wheel时报错,比如安装torch-scatter显示Error building wheel for torch-scatter
和一大串错误信息。
解决方法是手动下载whl文件,再安装。
官方文档上给出了支持的pytorch和cuda版本:https://data.pyg.org/whl/
在这里找到自己版本对应的,比如我就是torch-1.9.1+cu111
然后找到每个包对应python版本和系统型号下载。根据官方文档,torch-cluster和torch-spline-conv是可选的,其他包都要装一下。
下载后,进入放.whl文件的目录下,执行
pip install torch_scatter-2.0.9-cp38-cp38-win_amd64.whl
每个下载好的whl文件安装好后,执行
pip install torch_geometric
就OK啦!