本文主要是介绍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即可安装成功
这篇关于MAST中安装spatial-correlation-sampler踩坑记录的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!