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

你心

于 2020-12-03 11:43:19 发布

阅读量1.8k

点赞数

文章标签：深度学习

本文链接：https://blog.csdn.net/weixin_44350541/article/details/110524266

版权

博主记录了配置MAST时安装spatial-correlation-sampler遇到的问题及解决方法。pip安装后check.py测试失败，手动安装第一步出错，原因可能是torch与CUDA版本不适配。重新配置环境，包括python、CUDA、gcc、torch及torchvision版本，最终顺利安装并通过测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

论文：《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。