SOLOv2 INSTALL error

[gpsherry]

Hi，我在torch1.5、torch1.4上install SOLOV2都报这个类似错误，请问是什么原因呢？

[miyukiKawa]

If you replace all "AT_CHECK" with "TORCH_CHECK", you will be able to compile.

[rfezzani]

I solved this by installing specific versions of pytorch, nvcc and g++ in the conda environment:

conda create -n solo python=3.7 -y
conda activate solo
conda install pytorch==1.4.0 torchvision==0.5.0 -c pytorch -y
conda install -c conda-forge cudatoolkit-dev -y
conda install gxx_linux-64 -y
pip install -ve .

:wink:

[jiangfeng94]

If you replace all "AT_CHECK" with "TORCH_CHECK", you will be able to compile.

Successfully installed. by replace all "AT_CHECK" with "TORCH_CHECK"