PyG的安装
很早就想了解一下图神经网络,终于有时间学习一下了,下面记录一下安装 PyG 的过程。
PyG GitHub官网地址:GitHub - pyg-team/pytorch_geometric: Graph Neural Network Library for PyTorch
这个官网我觉得很好的一点是他一直在更新,而且基本上所有的图神经网络模型都嵌在里面,下载这一个可以体验所有模型,童叟无欺啊~~~
在安装 PyG 时,不要直接安装 pip install ,这样会有问题。
我们首先要安装一些依赖,轮子地址:data.pyg.org/whl/,从这里面我们根据自己 PyTorch、Python 和 cuda 版本选择正确的选项。
如何查看 PyTorch、Python 和 cuda 版本:
打开 Anaconda 的 Prompt (小黑窗口),激活你需要下载到的环境,之后通过 conda list 可以看到你环境中安装的所有包的信息。
看完版本之后,我们去轮子的官网去找对应的选项。
点进去之后就会看到下列图片,除了 Pyg.lib 之外,还有四个依赖包,我们要做的就是安装他们!
这里,cp 表示的是 Python 的版本,而 Linux 和 Win 就是你的系统啦,找到合适的就可以点击下载啦!
等待下载完成
下载完成之后,还记得前面要记住的两个框吗,现在就是用到他们的时候啦!
如果你直接 pip 安装,会出现不存在这个文件,导致安装失败。
所有,在这里,你的环境后面的地址是什么,你就把刚刚下载的四个轮子放到那个地址下面!
因为我不想放在 C 盘里面,所有我就选在环境所在的文件夹下,这样也好找!之后就可以一个一个 pip啦~~~
这四个轮子装完之后,就可以装 PyG 啦!输入下列代码:
pip install torch_geometric -i https://pypi.tuna.tsinghua.edu.cn/simple/
之后我们再去 conda list 看一下,就出现了这些包,这就表示安装完成啦!