最近一直在做图神经网络的相关项目,主要在用PyG的相关依赖在做。但是PyG并不像Pytorch那么友好,为了方便以后配置环境写一篇博客记录一下。
一般企业的服务器是不能够连接外网,或者只能在各种限制条件下连接外网,所以会带来很大的不便。本教程亲测可行,大多数采用离线安装的方式,安全快捷。以下操作以我已经成功安装的torch1.9.0 和 pyg1.7.2 为例。(注:网上大多教程都推荐torch版本安装x.x.0,即最后一个版本号为0与pyg最适配。)
第一步 安装pytorch离线安装办法:
就像cuda版本与pytorch版本有对应一样。 Pyg与pytorch也是有版本对应要求。所以在安装PyG之前确定Pytorch的版本,如果不符合要求还要卸载重装。
1) 来到torch官网下载你需要的相关包,如下图所示,选择适配你的系统和cuda版本的torch下载。
https://download.pytorch.org/whl/torch_stable.html
2)进入要安装torch和pyg的环境:conda activate xxx
3)离线安装:pip install /x/x/torch.xxxxx.whl
4)安装成功后,百度一下与1.9.0适配的torchvision和torchaudio版本号。再下载与torch版本匹配的torchvision和torchaudio。如下图:
第二步 安装PyG离线安装:
1)点击进入下面网站,https://pytorch-geometric.com/whl/。找到自己的torch和cuda对应的依赖,本人的torch版本是1.9.0,cuda版本是cu102,如下所示。
点击torch-1.8.0+cu102进入
2)依次安装对应系统和版本的四个PyG依赖,并使用pip install 离线安装 。
torch-cluster
torch-scatter
torch-sparse
torch-spline-conv
3)最后使用 pip install torch_geometric==1.7.2 安装最终的PyG即可使用。
本文章还参考了以下内容:
安装PyG:
PyTorch Geometric (PyG) 库的安装_steelDK的博客-CSDN博客
安装Pytorch:
Pytorch 离线安装教程( Win10 )_pytorch离线安装-CSDN博客
补充:
Installation — pytorch_geometric documentation 可以在该网站 查询CUDA Pyroch PyG的版本对应关系