在LINUX服务器上配置Pytorch以及GNN的环境(pytorch_geometric)

最近一直在做图神经网络的相关项目,主要在用PyG的相关依赖在做。但是PyG并不像Pytorch那么友好,为了方便以后配置环境写一篇博客记录一下。

一般企业的服务器是不能够连接外网,或者只能在各种限制条件下连接外网,所以会带来很大的不便。本教程亲测可行,大多数采用离线安装的方式,安全快捷。以下操作以我已经成功安装的torch1.9.0pyg1.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:

PyG的安装_pyg安装-CSDN博客

PyTorch Geometric (PyG) 库的安装_steelDK的博客-CSDN博客

安装Pytorch:

Pytorch 离线安装教程( Win10 )_pytorch离线安装-CSDN博客

补充:

Installation — pytorch_geometric documentation 可以在该网站 查询CUDA Pyroch PyG的版本对应关系

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值