torch_geometric安装踩坑

本文主要是介绍torch_geometric安装踩坑，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一定要按照官网的安装步骤来，否则安装完再导入包时很可能出现DLL链接错误。
具体安装步骤：
首先命令行查看自己电脑安装的pytorch和cuda版本：

python -c "import torch; print(torch.__version__)"
python -c "import torch; print(torch.version.cuda)"

之后下载对应版本的torch_geometric四个依赖包：

pip install torch-scatter -f https://data.pyg.org/whl/torch-1.9.1+cu102.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.9.1+cu102.html
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.9.1+cu102.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.9.1+cu102.html

最后安装torch-geometric：

pip install torch-geometric

官网链接：
https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

这篇关于torch_geometric安装踩坑的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---