MAST中安装spatial-correlation-sampler踩坑记录

论文:《MAST: A Memory-Augmented Self-Supervised Tracker》
github地址:https://github.com/zlai0/MAST

记录一下配置MAST中的安装spatial-correlation-sampler过程中的坑,几个月前跑过这个网络,最近换了个服务器重新配置环境,spatial-correlation-sampler的安装过程遇到了点小问题,在此mark下。

问题:pip install spatial-correlation-sampler后python check.py forward一直fail,故手动从git spatial-correlation-sampler进行安装,按照github上的安装指示进行安装,第一步就遇到下面的错误。在这里插入图片描述
解决:可能是当前虚拟环境下的torch与当前CUDA版本不适配导致的。故重新配置环境:python3.7,CUDA10.0,gcc升级到了7.5,torch1.1.0(CUDA10.0版本),torchvision0.3.0。并重新pip install -r requirements.txt,spatial-correlation-sampler也一并安装好了,并顺利通过了check.py。

注:CUDA10.0版本对应torch及torchvision下载链接:
https://download.pytorch.org/whl/cu100/torch_stable.html
方法:先wget到服务器下,再pip install whl即可安装成功

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值