pytorch-geometric安装

1.确定PyTorch版本

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

以我的电脑为例：

我的troch环境是1.8.0+cpu，这个参数很重要，后面pytorch-geometric安装要用到

如果你的torch是GPU版本，那么找到cuda版本，比如：cu110

GPU版本的torch查看cuda版本：

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

2. 下载pytorch-geometric依赖包

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

${TORCH}   ${CUDA} 替换成第一步查看到的torch版本和cuda版本，比如，我的torch是 1.8.0+cpu ，那么我的安装命令就是

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch--1.8.0+cpu.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch--1.8.0+cpu.html
pip install torch-geometric

如果你的torch是gpu版本的，那么同理，将上面的cpu换成‘cuxxx’（举个例子，实际换成什么要看你自己的cuda版本，比如cu110）

到这儿就完成啦，可以开始做简单的图神经网络任务了

 

参考：https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html