30系列显卡安装spatial

CUDA11.0安装spatial_correlation_sampler

配置：
显卡：3070
cuda11.0
torch 1.7.0

安装spatial_correlation_sampler对于CUDA<11.0 并且 torch<1.7 的机器来说可以直接

pip install spatial_correlation_sampler

但是对于30系列显卡CUDA版本必须要>=11.0并且torch>=1.7.0的机器上面的pip就不行了。作者已经做了一些修改，但还没能直接pip install ，所以我们只能通过他提供的源码进行编译。
链接: 点击这里.

然后编译setup.py文件就行了，如下命令行。

python setup.py install

参考：
https://github.com/ClementPinard/Pytorch-Correlation-extension/issues/52