The pynvml package is deprecated. Please install nvidia-ml-py instead

When running python demo.py --model_path lingbot_map/lingbot-map-long.pt --image_folder example/courthouse --mask_sky , I get the following warning: FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you.