PyG的安装

Ray-Z / 2024-09-26 / 原文

PyG的安装

很早就想了解一下图神经网络,终于有时间学习一下了,下面记录一下安装 PyG 的过程。

PyG GitHub官网地址:GitHub - pyg-team/pytorch_geometric: Graph Neural Network Library for PyTorch

这个官网我觉得很好的一点是他一直在更新,而且基本上所有的图神经网络模型都嵌在里面,下载这一个可以体验所有模型,童叟无欺啊~~~

image-20240926092404176

image-20240926092835492

在安装 PyG 时,不要直接安装 pip install ,这样会有问题。

image-20240926093114355

我们首先要安装一些依赖,轮子地址:data.pyg.org/whl/,从这里面我们根据自己 PyTorch、Python 和 cuda 版本选择正确的选项。

如何查看 PyTorch、Python 和 cuda 版本:

打开 Anaconda 的 Prompt (小黑窗口),激活你需要下载到的环境,之后通过 conda list 可以看到你环境中安装的所有包的信息。

image-20240926093821573

image-20240926094523538

image-20240926094607006

看完版本之后,我们去轮子的官网去找对应的选项。

image-20240926094854801

点进去之后就会看到下列图片,除了 Pyg.lib 之外,还有四个依赖包,我们要做的就是安装他们!

这里,cp 表示的是 Python 的版本,而 Linux 和 Win 就是你的系统啦,找到合适的就可以点击下载啦!

image-20240926095042452

等待下载完成

联想截图_20240926090600

下载完成之后,还记得前面要记住的两个框吗,现在就是用到他们的时候啦!

如果你直接 pip 安装,会出现不存在这个文件,导致安装失败。

image-20240926095728572

所有,在这里,你的环境后面的地址是什么,你就把刚刚下载的四个轮子放到那个地址下面!

因为我不想放在 C 盘里面,所有我就选在环境所在的文件夹下,这样也好找!之后就可以一个一个 pip啦~~~

image-20240926100215298

image-20240926100137020

这四个轮子装完之后,就可以装 PyG 啦!输入下列代码:

pip install torch_geometric -i https://pypi.tuna.tsinghua.edu.cn/simple/

image-20240926100421965

之后我们再去 conda list 看一下,就出现了这些包,这就表示安装完成啦!

image-20240926100522667