Cuda_runtime_api.h not in $CUDA_HOME/include

Environment

cuda-12.2
torch 2.1.0

Problem

when using pip install . to install local setup.py

I got an error /home/xingyu/anaconda3/envs/wm/lib/python3.8/site-packages/torch/include/ATen/cuda/CUDAContext.h:5:10: fatal error: cuda_runtime_api.h: No such file or directory

It appears that cuda_runtime_api.h is in

$CUDA_HOME/targets/x86_64-linux/include/thrust/system/cuda/detail/guarded_cuda_runtime_api.h
$CUDA_HOME/targets/x86_64-linux/include/cuda_runtime_api.h

instead of

$CUDA_HOME/include

Have anyone met the same problem?