最简单的安装方式:
pip install spatial_correlation_sampler
如果是天选之人,安装没有出错,那就恭喜!安装成功了。
但是大部分人可能会报错红红的一片屏幕,博主踩坑无数,终于成功,记录一下需要注意的点,希望可以帮那些踩坑的人快速解决问题。
这个包的官方地址如下:
https://github.com/ClementPinard/Pytorch-Correlation-extension/tree/fix_1.7
issue板块可能会有你需要的信息,可以看看别人的安装问题。
简单来说,安装这个包需要注意:
1.cuda11.0以上可能不可以直接使用pip install。需要把github上面的包下载下来解压,然后python setup.py install。(注意要下载fix_1.7分支,不是master。)
2.实测能安装的环境是不限的,不像有的帖子说必须cuda10.0+torch1.6.0。所以不必安装很多虚拟环境。博主的环境:win10+cuda11.1+torch1.9.0+vs2017
3.需要有VS2017-2019,其他版本不支持,且需要正确配置环境变量。可参考:
https://blog.csdn.net/qing666888/article/details/83246992
4.需要有GCC。安装最高版本即可,向下兼容的。
5.CUDA的环境变量需要正确配置
有报错可以逐一排查,不需要改他的代码,安装不了多半都是环境问题。