Open gpsherry opened 3 years ago
If you replace all "AT_CHECK" with "TORCH_CHECK", you will be able to compile.
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:
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"
Hi,我在torch1.5、torch1.4上install SOLOV2都报这个类似错误,请问是什么原因呢?