PyG快速安装（一键脚本，2021.7.14简单有效）

本文提供两种方法安装都可成功安装 pyg

通过脚本安装

直接通过脚本安装，自建一个python脚本，直接运行即可，代码如下：

import logging
import subprocess

from setuptools import setup

import torch

cuda_v = f"cu{torch.version.cuda.replace('.', '')}"
torch_v = torch.__version__.split('.')
torch_v = '.'.join(torch_v[:-1] + ['0'])


def system(command: str):
    output = subprocess.check_output(command, shell=True)
    logging.info(output)


system(f'pip install scipy')
system(f'pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install --no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install --no-index torch-spline-conv -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install torch-geometric')

离线本地安装

① 打开链接 官方资源下载地址

② 选择自己电脑安装的cuda版本

③ 点击进去后，对应自己的Python版本和系统，找到四个需要安装的文件：

torch_cluster
torch_scatter
torch_sparse
torch_spline_conv

比如说是windows，python3.7，那么选择如下：

torch_cluster-1.5.9-cp37-cp37m-win_amd64.whl
torch_scatter-2.0.6-cp37-cp37m-win_amd64.whl
torch_sparse-0.6.9-cp37-cp37m-win_amd64.whl
torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl

cp37表示python3.7.*的版本

④ 下载到本地任意文件夹之后，打开cmd，激活你的环境，pip install 即可

我下载到了：D:\下载文件\cuda安装包\

那么执行如下命令即可：

pip install D:\下载文件\cuda安装包\torch_cluster-1.5.9-cp37-cp37m-win_amd64.whl
pip install D:\下载文件\cuda安装包\torch_scatter-2.0.6-cp37-cp37m-win_amd64.whl
pip install D:\下载文件\cuda安装包\torch_sparse-0.6.9-cp37-cp37m-win_amd64.whl
pip install D:\下载文件\cuda安装包\torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl

安装结果测试

import torch
from torch_geometric.data import Data

edge_index = torch.tensor([[0, 1, 1, 2],
                           [1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)

data = Data(x=x, edge_index=edge_index)
print(data)

到这里基本上就测试完毕了。