torch_scatter安装

图卷积中常用的一个函数,直接pip install下载不下来,查阅各种资料之后跑通的一个比较方便的方法,记录一下:

1. 查看torch以及cuda版本:

import torch
print(torch.__version__)
print(torch.version.cuda)

2. 找版本都对应上的torch_scatter并下载:

https://pytorch-geometric.com/whl/

 

 

 cp是指python版本,可以在命令行中输入python查看。

例如我python版本为3.8,下载torch_scatter-2.0.9-cp38-cp38-win_amd64.whl

3. 安装:

在命令行中:

pip install 下载的whl文件所在路径

完成!


如果安装完成后import torch_scatter报错:

OSError: [WinError 127] 找不到指定的程序。

说明还是下载的版本不匹配。但按照上面流程找对应版本应该就能避免这个问题。

评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值