Could NOT find CUDA (missing: CUDA_CUDART_LIBRARY) (found version "12.1")
Try adding -DCUDA_CUDART_LIBRARY=/usr/local/cuda/lib64/libcudart.so when you run cmake command.
Try adding -DCUDA_CUDART_LIBRARY=/usr/local/cuda/lib64/libcudart.so when you run cmake command.