Really appreciate for your help, but I think the problem is not caused by apex. I removed the conda environment and clone the apex using
git clone GitHub - NVIDIA/apex: A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
and install it follow the commend as follow:
pip install -v --disable-pip-version-check --no-cache-dir --global-option="–cpp_ext" --global-option="–cuda_ext" ./
The error is still the same, but I found that, it is some csrc/cuda/***.cu file couldn’t find the THC, for example :SigmoidFocalLoss_cuda.cu deform_pool_cuda.cu ROIPool_cuda.cu. Maybe I can find a newer version of these function somewhere? Or any advice I can solve if.