[人工智能]ERROR: Failed building wheel for spatial-correlation-sampler

问题描述

安装spatial-correlation-sampler模块报错ERROR: Failed building wheel for spatial-correlation-sampler。

问题分析

出现该问题的原因大概率是pytorch版本或cuda版本与spatial-correlation-sampler要求不一致。
spatial-correlation-sampler要求版本：

This module is expected to compile for Pytorch 1.2, on Python > 3.5 and Python 2.7.

解决方案

建议新建一个pytorch1.2版本的环境，安装cuda10.0，再安装spatial-correlation-sampler模块。

1. 新建环境，pytorch1.2，python3.6

conda create -n pytorch_1.2 python=3.6   # 新建环境，pytorch1.2，python3.6

2. 激活环境

conda activate pytorch_1.2

3. 在该环境安装pytorch1.2及相应工具

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch 

4. 安装cuda10.0

地址: cuda10.0
请参考这篇文章: cuda10.0的安装教程

5. 在该环境中选择，cuda10.0版本

export CUDA_HOME=/usr/local/cuda-10.0

6. 安装spatial-correlation-sampler

pip install spatial-correlation-sampler==0.0.8

参考：

1. spatial-correlation-sampler
2. cuda10.0的安装教程