安装torch-scatter的坑

torch-scatter==1.4.0
torch-sparse==0.4.3
torch-cluster==1.4.5
torch-geometric==1.3.2
 在搭DSNet项目的环境时,遇到以上几个torch附属库的安装,分享一下安装时遇到的坑。

首先torch-scatter所依赖的CUDA版本基于你的Pytorch安装版本,可参考此文章。本文亲测成功的一个版本关系为Pytorch(1.1.0)+CUDA(9.0.176)+torch-scatter(1.4.0)+Ubuntu 16.04/windows 10。Pytorch安装命令行为

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

之后输入

pip install torch-scatter==1.4.0

 此时有可能出现以下错误:

 根据错误信息可知,与VS版本有关,其编译仅支持VS2012、2013、2015、2017版本,由于我安装有更高的2019版本,因此将该版本暂时卸载后即可安装成功。

 以此类推,其余几个torch库的版本也都亲测可以成功安装。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值