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

建议torch-geometric不要装最新版本,建议装1.2.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
    

    安装成功,结束!

  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值