Add pynvml dependency. pynvml is not required by any of the dependencies. However, we require this package for checking for GPU compatibility and some a function to get device memory used by merlin.io.Dataset.
Restricting pynvml to <11.5 due to an incompatibility between distributed<2023.2.1 and pynvml==11.5.0. If we at some point increase the min version of dask/distributed to 2023.2.1 then we can change the pynvml version restriction
Add pynvml dependency. pynvml is not required by any of the dependencies. However, we require this package for checking for GPU compatibility and some a function to get device memory used by
merlin.io.Dataset.Restricting pynvml to <11.5 due to an incompatibility between
distributed<2023.2.1andpynvml==11.5.0. If we at some point increase the min version of dask/distributed to 2023.2.1 then we can change the pynvml version restrictionFixes https://github.com/NVIDIA-Merlin/dataloader/issues/106