Ubuntu22.04 pytorch-geometric(PyG)安装

大菠菠菠萝

已于 2024-08-08 08:00:19 修改

阅读量748

点赞数 2

文章标签： pytorch python 深度学习

于 2023-05-31 17:22:34 首次发布

本文链接：https://blog.youkuaiyun.com/qq_44362147/article/details/130972541

版权

默认已经安装好了pytorch，如果没安装好，可参考博客：

pytorch库安装-优快云博客https://blog.youkuaiyun.com/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)