Spatial-Correlation-Sampler安装报错CHECK_CUDA(x)&CHECK_INPUT(x)

一、报错信息截图

二、源码安装过程

1. 这个包的原名是Pytorch-Correlation-extension Github地址
   历史版本点下图红框。

   

2. 下载到本地后解压

   tar -xzvf Pytorch-Correlation-extension-0.2.0.tar.gz 
   

3. 激活conda环境

   conda activate pytorch
   

4. [重要]修改报错位置的代码

   cd Pytorch-Correlation-extension-0.2.0/Correlation_Module
   sudo gedit correlation_sampler.cpp 
   

   将所有TORCH_CHECK更改为AT_CHECK（torch1.5以前的版本不支持TORCH_CHECK）
   

5. 编译与安装
   回到Pytorch-Correlation-extension-0.2.0目录下，使用setup.py安装

   cd ..
   python setup.py install
   

6. 安装成功

   pip list