默认已经安装好了pytorch,如果没安装好,可参考博客:
pytorch库安装-CSDN博客https://blog.csdn.net/qq_44362147/article/details/140346488?spm=1001.2014.3001.5501
pytorch版本>=1.13可以直接通过pip一步安装
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0+${CUDA}.html
以pytorch=2.0.0 cuda=11.8为例,pip命令:
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0+cu118.html
${CUDA} 根据cuda版本修改为cu118。
torch2.0支持的cuda或cpu版本
torch1.13支持的cuda或cpu版本
其他Pytorch版本
需要先安装官网给出的五个包,然后在安装pytorch_geometric。
首先在 https://data.pyg.org/whl/ 中下载轮子,找到你的torch和cuda版本。以torch1.12.1+cuda116为例。
pt为pytorch版本,cu为cuda版本,cp为虚拟环境中python版本,linux操作系统。
进入你的虚拟环境:
conda activate 虚拟环境名字
pip安装:
pip install 下载路径
例如:
pip install /home/guest/torch_cluster-1.6.0+pt112cu116-cp310-cp310-linux_x86_64.whl
pip install /home/guest/pyg_lib-0.2.0+pt112cu116-cp310-cp310-linux_x86_64.whl
pip install /home/guest/torch_scatter-2.1.0+pt112cu116-cp310-cp310-linux_x86_64.whl
pip install /home/guest/torch_sparse-0.6.16+pt112cu116-cp310-cp310-linux_x86_64.whl
pip install /home/guest/torch_spline_conv-1.2.1+pt112cu116-cp310-cp310-linux_x86_64.whl
所有安装包全部安装完成后:
pip install torch-geometric
官方网址:
pyg-team/pytorch_geometric: Graph Neural Network Library for PyTorch (github.com)