Hey guys,
I found a solution to this problem. You can download NVTX3 from GitHub - NVIDIA/NVTX: The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resources in your applications.. Then, the directory pathway in libtorch/share/cmake/Caffe2/public/cuda.cmake should be adjusted correspondingly. Once you have the NVTX3, the warning will not appear.
Best, Weitao.