官网的简介

Installation — pytorch_geometric documentation (pytorch-geometric.readthedocs.io)

按照以下的安装步骤来即可

1.查看torch版本

        python -c "import torch; print(torch.__version__)"

2.下载相关的包 : data.pyg.org/whl/  

        下载 这几个包 torch_scatter   torch_sparse    torch_cluster  torch_cluster

       对应下载版本的包的目录

        

从下面页面下载四个包,  cp36表示在python3.6版本环境里面可以安装的,cp37,cp38同理要对应python版本,  linux和win分别表示Linux系统和windows系统,下载对应的包即可

3.下载完成后,按照以下顺序安装

        3.1先安装torch_scatter   :   

pip install   torch_scatter的路径(.....whl)

      3.2    安装torch_sparse

     3.3    安装torch_cluster

    3.4 安装torch_spline_conv

  4.安装完成上面四个包之后,运行

pip install torch-geometric

(这里直接复制站贴可能会失败,建议自己敲键盘)

  5.测试  

 import torch_geometric (无报错即成功)

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐