Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add pynvml dependency #237

Merged
merged 3 commits into from
Mar 8, 2023

Conversation

oliverholworthy
Copy link
Member

@oliverholworthy oliverholworthy commented Mar 8, 2023

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

Fixes NVIDIA-Merlin/dataloader#106

@oliverholworthy oliverholworthy added the chore Maintenance for the repository label Mar 8, 2023
@oliverholworthy oliverholworthy self-assigned this Mar 8, 2023
@oliverholworthy oliverholworthy added this to the Merlin 23.03 milestone Mar 8, 2023
@karlhigley karlhigley merged commit aad0c87 into NVIDIA-Merlin:main Mar 8, 2023
oliverholworthy added a commit that referenced this pull request Mar 10, 2023
* Add pynvml dependency

Correct name of pynvml package

* Add comment to requirements.txt about pynvml version

* Remove whitespace from requirements.txt
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
chore Maintenance for the repository
Projects
None yet
Development

Successfully merging this pull request may close these issues.

GPU is not detected properly when using SLURM
3 participants