python安装使用PyG图神经网络框架

首先切换到CMD，打开python交互环境

输入，查看你所安装的pytorch版本

import torch
print(torch.version)

我的当前pytorch版本是>>> 1.6.0

安装依赖包及torch-geometric

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-$TORCH+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-$TORCH+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-$TORCH+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-$TORCH+${CUDA}.html
pip install torch-geometric

在那里{CUDA}，并{TORCH}应通过特定CUDA版本（被替换cpu，cu92，cu101，cu102，cu110）和PyTorch版本（1.4.0，1.5.0，1.6.0，1.7.0）分别。例如，对于PyTorch 1.7.0和CUDA 11.0，键入：

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html
pip install torch-geometric

在安装torch-cluster报错，提示安装不上，解决办法如下：
1.打开下载网址为https://pytorch-geometric.com/whl/，选择相应的torch版，本人的torch版本为1.6.0+cpu

在这里插入图片描述

2. 同时本人的python版本为3.7.7，则选择对应的torch-cluster进行下载.whl文件进行本地安装。

在这里插入图片描述

3. 将上一步下载的.whl文件执行
   pip install 下载的.whl文件路径+.whl文件即可
4. 安装成功

检查安装结果

import torch
import torch_geometric

安装成功，结束！

有意者可加QQ：3271330538 还可点击666领取python资料