search

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.69k stars 307 forks source link

SOLOv2 INSTALL error #151

Open gpsherry opened 3 years ago

gpsherry commented 3 years ago
infoflow 2021-01-29 14-51-20

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

miyukiKawa commented 3 years ago

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

rfezzani commented 3 years ago

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 commented 3 years ago

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"