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

982 阅读1分钟
  1. 首先切换到CMD,打开python交互环境

  2. 输入,查看你所安装的pytorch版本

    import torch
    print(torch.__version__)
    
    

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

  3. 安装依赖包及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. 安装成功

  4. 检查安装结果

    import torch
    import torch_geometric
    
    

    安装成功,结束!

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