Win 10 + cudn 10.1 + anaconda + pytorch 1.4.0+torch-scatter

在win 10 环境下配置torch-scatter,经历了很多坑,最后总结起来,就是要看官方文档,先给出官方文档的连接及相关有用的链接:

https://pytorch.org/

https://github.com/rusty1s/pytorch_geometric

https://github.com/pytorch/pytorch

在win 10环境下,cuda的版本只能如下所示:

因此需要安装pytorch 1.4.0版本,cuda 为10.1版本的,其他的版本不支持。

接下来输入如下的命令安装:

pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html  

pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install torch-geometric

${CUDA}换为cu101,可成功安装

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值